/external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
H A D | 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" 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( 47 xla [all...] |
H A D | 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 [all...] |
H A D | 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 [all...] |
H A D | 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,
|
H A D | 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 [all...] |
H A D | 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,
|
H A D | 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 [all...] |
H A D | 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 [all...] |
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
H A D | gpu_constants.cc | 16 #include "tensorflow/compiler/xla/service/gpu/gpu_constants.h" 18 namespace xla { namespace 25 } // namespace xla
|
H A D | gpu_constants.h | 19 #include "tensorflow/compiler/xla/types.h" 21 namespace xla { namespace 29 } // namespace xla
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
H A D | xla_helpers.h | 22 #include "tensorflow/compiler/xla/client/computation_builder.h" 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, 53 static xla [all...] |
H A D | 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 [all...] |
H A D | 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 [all...] |
H A D | shape_util.h | 21 #include "tensorflow/compiler/xla/xla_data.pb.h" 29 Status XLAShapeToTensorShape(const xla::Shape& shape, 36 xla::Shape* shape);
|
H A D | 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/interpreter/ |
H A D | 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 [all...] |
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
H A D | reduction_ops.h | 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( 57 xla [all...] |
H A D | gather_op_helpers.h | 22 #include "tensorflow/compiler/xla/client/client_library.h" 23 #include "tensorflow/compiler/xla/client/computation_builder.h" 36 Status XlaGather(const xla::ComputationDataHandle& input, 38 const xla::ComputationDataHandle& indices, 41 xla::ComputationBuilder* builder, 42 xla::ComputationDataHandle* gather_output);
|
H A D | reduction_ops.cc | 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 { 63 xla [all...] |
H A D | cwise_ops.h | 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 58 virtual xla::ComputationDataHandle Computation( 59 XlaOpKernelContext* ctx, const xla::ComputationDataHandle& lhs, 61 const xla::ComputationDataHandle& rhs, 70 static std::pair<xla::ComputationDataHandle, xla::ComputationDataHandle> 71 Broadcast(xla::ComputationBuilder* builder, 72 const xla [all...] |
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
H A D | 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 [all...] |
H A D | 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 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, 100 xla [all...] |
/external/tensorflow/tensorflow/compiler/xla/ |
H A D | execution_options_util.h | 19 #include "tensorflow/compiler/xla/xla.pb.h" 21 namespace xla { namespace 27 } // namespace xla
|
H A D | status.h | 22 namespace xla { namespace 26 } // namespace xla
|
/external/tensorflow/tensorflow/compiler/xla/legacy_flags/ |
H A D | debug_options_flags.h | 21 #include "tensorflow/compiler/xla/xla.pb.h" 24 namespace xla { namespace 33 xla::DebugOptions GetDebugOptionsFromFlags(); 36 } // namespace xla
|