Searched refs:xla (Results 1 - 25 of 814) sorted by relevance

1234567891011>>

/external/tensorflow/tensorflow/compiler/tf2xla/lib/
H A Dscatter.h21 #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 Dtriangular_solve.h19 #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 Dutil.h19 #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 Dcholesky.h19 #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 Dwhile_loop.h22 #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 Dbatch_dot.h19 #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 Dutil.cc21 #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 Dwhile_loop.cc18 #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 Dgpu_constants.cc16 #include "tensorflow/compiler/xla/service/gpu/gpu_constants.h"
18 namespace xla { namespace
25 } // namespace xla
H A Dgpu_constants.h19 #include "tensorflow/compiler/xla/types.h"
21 namespace xla { namespace
29 } // namespace xla
/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_helpers.h22 #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 Dtype_util.cc23 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 Dsharding_util.h20 #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 Dshape_util.h21 #include "tensorflow/compiler/xla/xla_data.pb.h"
29 Status XLAShapeToTensorShape(const xla::Shape& shape,
36 xla::Shape* shape);
H A Dtf2xla.h20 #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 Dinterpreter_transfer_manager.cc16 #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 Dreduction_ops.h22 #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 Dgather_op_helpers.h22 #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 Dreduction_ops.cc22 #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 Dcwise_ops.h22 #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 Dcpu_runtime.cc16 #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 Dcpu_runtime.h18 // 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 Dexecution_options_util.h19 #include "tensorflow/compiler/xla/xla.pb.h"
21 namespace xla { namespace
27 } // namespace xla
H A Dstatus.h22 namespace xla { namespace
26 } // namespace xla
/external/tensorflow/tensorflow/compiler/xla/legacy_flags/
H A Ddebug_options_flags.h21 #include "tensorflow/compiler/xla/xla.pb.h"
24 namespace xla { namespace
33 xla::DebugOptions GetDebugOptionsFromFlags();
36 } // namespace xla

Completed in 321 milliseconds

1234567891011>>