| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| scatter.h | 21 #include "tensorflow/compiler/xla/client/computation.h"
22 #include "tensorflow/compiler/xla/client/computation_builder.h"
23 #include "tensorflow/compiler/xla/statusor.h"
27 // Builds an XLA computation that performs a scatter operation on `buffer`,
42 xla::StatusOr<xla::ComputationDataHandle> XlaScatter(
43 const xla::ComputationDataHandle& buffer,
44 const xla::ComputationDataHandle& updates,
45 const xla::ComputationDataHandle& indices, bool indices_are_vectors,
46 const std::function<xla::ComputationDataHandle
[all...] |
| triangular_solve.h | 19 #include "tensorflow/compiler/xla/client/computation.h"
20 #include "tensorflow/compiler/xla/client/computation_builder.h"
60 xla::StatusOr<xla::ComputationDataHandle> TriangularSolve(
61 xla::ComputationBuilder* builder, const xla::ComputationDataHandle& a,
62 xla::ComputationDataHandle b, bool left_side, bool lower, bool transpose_a,
65 xla::StatusOr<xla::ComputationDataHandle> TriangularSolveLeftLooking(
66 xla::ComputationBuilder* builder, const xla::ComputationDataHandle& a
[all...] |
| util.h | 19 #include "tensorflow/compiler/xla/client/computation.h"
20 #include "tensorflow/compiler/xla/client/computation_builder.h"
21 #include "tensorflow/compiler/xla/statusor.h"
27 xla::ComputationDataHandle Zeros(xla::ComputationBuilder* builder,
28 const xla::Shape& shape);
32 xla::ComputationDataHandle FloatLiteral(xla::ComputationBuilder* builder,
33 xla::PrimitiveType type, double value);
37 xla::ComputationDataHandle IntegerLiteral(xla::ComputationBuilder* builder
[all...] |
| cholesky.h | 19 #include "tensorflow/compiler/xla/client/computation.h"
20 #include "tensorflow/compiler/xla/client/computation_builder.h"
33 xla::StatusOr<xla::ComputationDataHandle> Cholesky(
34 xla::ComputationBuilder* builder, xla::ComputationDataHandle a,
|
| while_loop.h | 22 #include "tensorflow/compiler/xla/client/computation.h"
23 #include "tensorflow/compiler/xla/client/computation_builder.h"
24 #include "tensorflow/compiler/xla/statusor.h"
32 typedef std::function<xla::StatusOr<xla::ComputationDataHandle>(
33 gtl::ArraySlice<xla::ComputationDataHandle>, xla::ComputationBuilder*)>
38 typedef std::function<xla::StatusOr<std::vector<xla::ComputationDataHandle>>(
39 gtl::ArraySlice<xla::ComputationDataHandle>, xla::ComputationBuilder*)
[all...] |
| batch_dot.h | 19 #include "tensorflow/compiler/xla/client/computation.h"
20 #include "tensorflow/compiler/xla/client/computation_builder.h"
46 xla::StatusOr<xla::ComputationDataHandle> BatchDot(
47 xla::ComputationBuilder* builder, xla::ComputationDataHandle x,
48 xla::ComputationDataHandle y, bool transpose_x, bool transpose_y,
|
| util.cc | 21 #include "tensorflow/compiler/xla/literal_util.h"
22 #include "tensorflow/compiler/xla/shape_util.h"
23 #include "tensorflow/compiler/xla/status_macros.h"
24 #include "tensorflow/compiler/xla/statusor.h"
25 #include "tensorflow/compiler/xla/util.h"
30 xla::ComputationDataHandle Zeros(xla::ComputationBuilder* builder,
31 const xla::Shape& shape) {
33 builder->ConstantLiteral(xla::Literal::Zero(shape.element_type())),
34 xla::AsInt64Slice(shape.dimensions()))
[all...] |
| while_loop.cc | 18 #include "tensorflow/compiler/xla/shape_util.h"
19 #include "tensorflow/compiler/xla/status_macros.h"
23 xla::StatusOr<std::vector<xla::ComputationDataHandle>> XlaWhileLoop(
26 gtl::ArraySlice<xla::ComputationDataHandle> initial_values,
27 StringPiece name, xla::ComputationBuilder* builder) {
29 std::vector<xla::Shape> var_shapes;
31 for (const xla::ComputationDataHandle& input : initial_values) {
35 xla::Shape tuple_shape = xla::ShapeUtil::MakeTupleShape(var_shapes)
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/ |
| execution_options_util.h | 19 #include "tensorflow/compiler/xla/xla.pb.h"
21 namespace xla { namespace
27 } // namespace xla
|
| status.h | 22 namespace xla { namespace
26 } // namespace xla
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| gpu_constants.cc | 16 #include "tensorflow/compiler/xla/service/gpu/gpu_constants.h"
18 namespace xla { namespace
25 } // namespace xla
|
| gpu_constants.h | 19 #include "tensorflow/compiler/xla/types.h"
21 namespace xla { namespace
29 } // namespace xla
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| xla_helpers.h | 16 // This file defines helper routines for the XLA device.
22 #include "tensorflow/compiler/xla/client/computation_builder.h"
28 // Helper methods for building XLA computations.
33 static xla::ComputationDataHandle MinValue(xla::ComputationBuilder* b,
38 static xla::ComputationDataHandle MaxValue(xla::ComputationBuilder* b,
43 static xla::ComputationDataHandle Zero(xla::ComputationBuilder* b,
48 static xla::ComputationDataHandle One(xla::ComputationBuilder* b
[all...] |
| type_util.cc | 23 Status DataTypeToPrimitiveType(DataType data_type, xla::PrimitiveType* type) {
26 *type = xla::PRED;
29 *type = xla::S8;
32 *type = xla::S16;
35 *type = xla::S32;
38 *type = xla::S64;
41 *type = xla::U8;
44 *type = xla::U16;
47 *type = xla::U32;
50 *type = xla::U64
[all...] |
| sharding_util.h | 20 #include "tensorflow/compiler/xla/client/sharding_builder.h"
21 #include "tensorflow/compiler/xla/status_macros.h"
35 // - a sharding set as per xla::sharding_builder::AssignDevice.
36 xla::StatusOr<tensorflow::gtl::optional<xla::OpSharding>>
38 tensorflow::gtl::optional<xla::OpSharding>
41 xla::StatusOr<tensorflow::gtl::optional<xla::OpSharding>>
44 xla::StatusOr<tensorflow::gtl::optional<xla::OpSharding>
[all...] |
| shape_util.h | 16 // Utilities for working with XLA shapes.
21 #include "tensorflow/compiler/xla/xla_data.pb.h"
27 // Convert an XLA Shape into the equivalent TensorFlow shape. May fail since
28 // not all XLA shapes can be represented as TensorShapes.
29 Status XLAShapeToTensorShape(const xla::Shape& shape,
32 // Convert a TensorShape into the equivalent XLA Shape proto. Unlike Tensorflow,
33 // XLA shapes include the type. Not all `dtype` values can be represented by
34 // XLA, so this conversion may fail.
36 xla::Shape* shape);
|
| tf2xla.h | 20 #include "tensorflow/compiler/xla/client/client.h"
21 #include "tensorflow/compiler/xla/client/computation.h"
26 // Converts a tensorflow::GraphDef into an xla::Computation. The given `config`
34 const tf2xla::Config& config, xla::Client* client,
35 xla::Computation* computation);
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_runtime.h | 18 // xla::cpu::SimpleResolver. It also defines a per-CpuExecutable context
29 #include "tensorflow/compiler/xla/service/cpu/xfeed_manager.h"
30 #include "tensorflow/compiler/xla/types.h"
32 namespace xla { namespace
39 // third_party/tensorflow/compiler/xla/service/cpu/simple_orc_jit.cc) maps
59 // All symbol names for XLA CPU runtime functions need to start with this
68 } // namespace xla
84 xla::int32 buffer_length, const void* shape, xla::int32 shape_length);
99 xla::int32 buffer_length, void* buffer_ptr, const void* shape_ptr
[all...] |
| cpu_runtime.cc | 16 #include "tensorflow/compiler/xla/service/cpu/cpu_runtime.h"
20 #include "tensorflow/compiler/xla/service/llvm_ir/llvm_util.h"
25 namespace xla { namespace
65 } // namespace xla
69 tensorflow::string ShapeString(const void* shape_ptr, xla::int32 shape_length) {
70 xla::StatusOr<xla::Shape> shape =
71 xla::llvm_ir::DecodeSelfDescribingShapeConstant(shape_ptr, shape_length);
73 return xla::ShapeUtil::HumanStringWithLayout(shape.ValueOrDie());
80 void* __xla_cpu_runtime_AcquireInfeedBufferForDequeue(xla::int32 buffer_length
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| interpreter_transfer_manager.cc | 16 #include "tensorflow/compiler/xla/service/interpreter/interpreter_transfer_manager.h"
20 #include "tensorflow/compiler/xla/ptr_util.h"
21 #include "tensorflow/compiler/xla/service/interpreter/platform_id.h"
22 #include "tensorflow/compiler/xla/service/transfer_manager.h"
26 namespace xla { namespace
32 } // namespace xla
34 static std::unique_ptr<xla::TransferManager>
36 return xla::MakeUnique<xla::InterpreterTransferManager>();
40 xla::TransferManager::RegisterTransferManager
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| reduction_ops.h | 16 // XLA-specific base classes for Reduction Ops.
22 #include "tensorflow/compiler/xla/client/computation_builder.h"
31 // xla::ComputationBuilder and BuildFinalizer adds the
33 // xla::ComputationBuilder.
40 virtual xla::ComputationDataHandle InitialValue(
41 xla::ComputationBuilder* builder);
47 virtual void BuildReducer(xla::ComputationBuilder* builder,
48 const xla::ComputationDataHandle& scalar_lhs,
49 const xla::ComputationDataHandle& scalar_rhs) = 0;
56 virtual xla::ComputationDataHandle BuildFinalizer
[all...] |
| gather_op_helpers.h | 16 // Helper methods for XLA Gather Ops.
22 #include "tensorflow/compiler/xla/client/client_library.h"
23 #include "tensorflow/compiler/xla/client/computation_builder.h"
29 // Adds to builder an XLA computation that performs a gather on input (of
36 Status XlaGather(const xla::ComputationDataHandle& input,
38 const xla::ComputationDataHandle& indices,
41 xla::ComputationBuilder* builder,
42 xla::ComputationDataHandle* gather_output);
|
| reduction_ops.cc | 16 // XLA-specific reduction Ops.
22 #include "tensorflow/compiler/xla/literal_util.h"
31 void BuildReducer(xla::ComputationBuilder* builder,
32 const xla::ComputationDataHandle& scalar_lhs,
33 const xla::ComputationDataHandle& scalar_rhs) override {
44 xla::ComputationDataHandle InitialValue(
45 xla::ComputationBuilder* builder) override {
49 void BuildReducer(xla::ComputationBuilder* builder,
50 const xla::ComputationDataHandle& scalar_lhs,
51 const xla::ComputationDataHandle& scalar_rhs) override
[all...] |
| cwise_ops.h | 16 // XLA-specific base classes for Unary and Binary Ops.
22 #include "tensorflow/compiler/xla/client/client_library.h"
23 #include "tensorflow/compiler/xla/client/computation_builder.h"
33 // implementation of the operation to a xla::ComputationBuilder. For most
34 // arithmetic Ops XLA handles the broadcasting automatically given the input
52 // may be useful for Ops that can't use standard XLA automatic
57 // in the XLA documentation.
58 virtual xla::ComputationDataHandle Computation(
59 XlaOpKernelContext* ctx, const xla::ComputationDataHandle& lhs,
61 const xla::ComputationDataHandle& rhs
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/legacy_flags/ |
| debug_options_flags.h | 21 #include "tensorflow/compiler/xla/xla.pb.h"
24 namespace xla { namespace
33 xla::DebugOptions GetDebugOptionsFromFlags();
36 } // namespace xla
|
