| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| type_util.h | 19 #include "tensorflow/compiler/xla/xla_data.pb.h"
25 // Converts a Tensorflow DataType to an XLA PrimitiveType.
26 Status DataTypeToPrimitiveType(DataType data_type, xla::PrimitiveType* type);
|
| literal_util.h | 16 // Utilities for working with XLA Literals.
21 #include "tensorflow/compiler/xla/literal_util.h"
22 #include "tensorflow/compiler/xla/xla_data.pb.h"
28 // Copies 'host_tensor' to an XLA Literal. Fails if host_tensor is of an
30 Status HostTensorToLiteral(const Tensor& host_tensor, xla::Literal* literal);
37 // to the same XLA type (e.g. INT32 and QINT32 both map to INT32 in
38 // XLA).
39 Status LiteralToHostTensor(const xla::Literal& literal, DataType target_type,
45 Status CopyLiteralToHostTensor(const xla::Literal& literal,
|
| xla_helpers.cc | 16 // This file defines helper routines for XLA compilation.
25 #include "tensorflow/compiler/xla/client/computation_builder.h"
26 #include "tensorflow/compiler/xla/types.h"
34 Status ArgMinMax(xla::ComputationBuilder* builder, XlaOpKernelContext* ctx,
35 const xla::ComputationDataHandle& input,
38 xla::ComputationDataHandle* argminmax) {
39 xla::ComputationDataHandle init_value;
40 const xla::Computation* reducer;
49 xla::PrimitiveType xla_output_type;
52 xla::ComputationDataHandle input_max = builder->Reduce
[all...] |
| sharding_util.cc | 29 xla::StatusOr<tensorflow::gtl::optional<xla::OpSharding>>
32 return tensorflow::gtl::optional<xla::OpSharding>();
35 xla::OpSharding sharding;
38 return xla::InvalidArgument(
40 "xla::OpSharding proto.");
42 return tensorflow::gtl::optional<xla::OpSharding>(sharding);
52 xla::StatusOr<tensorflow::gtl::optional<xla::OpSharding>>
55 tensorflow::gtl::optional<xla::OpSharding> explicit_sharding)
[all...] |
| xla_resource.h | 21 #include "tensorflow/compiler/xla/client/computation_builder.h"
22 #include "tensorflow/compiler/xla/xla_data.pb.h"
41 const xla::ComputationDataHandle& initial_value,
72 const xla::ComputationDataHandle& value() const { return value_; }
76 const xla::ComputationDataHandle& initial_value() const {
89 Status SetValue(const xla::ComputationDataHandle& value);
92 Status SetZeroValue(xla::ComputationBuilder* builder);
99 xla::ComputationBuilder* builder,
102 // Packs a resource into a single XLA value `pack`, suitable for use as
107 Status Pack(xla::ComputationDataHandle* pack
[all...] |
| literal_util.cc | 20 #include "tensorflow/compiler/xla/literal_util.h"
25 Status HostTensorToLiteral(const Tensor& host_tensor, xla::Literal* literal) {
26 xla::Shape literal_shape;
30 *literal = xla::Literal(literal_shape);
43 Status CopyLiteralToHostTensor(const xla::Literal& literal,
45 TF_RET_CHECK(xla::ShapeUtil::IsArray(literal.shape()) &&
46 xla::ShapeUtil::ElementsIn(literal.shape()) ==
48 xla::PrimitiveType primitive_type;
54 xla::PrimitiveType_Name(literal.shape().element_type()),
66 Status LiteralToHostTensor(const xla::Literal& literal, DataType target_type
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/ |
| execution_options_util.cc | 15 #include "tensorflow/compiler/xla/execution_options_util.h"
16 #include "tensorflow/compiler/xla/legacy_flags/debug_options_flags.h"
18 namespace xla { namespace
27 } // namespace xla
|
| text_literal_writer.h | 19 #include "tensorflow/compiler/xla/literal_util.h"
20 #include "tensorflow/compiler/xla/types.h"
21 #include "tensorflow/compiler/xla/xla_data.pb.h"
26 namespace xla { namespace
37 // This should be readable by xla::TextLiteralReader.
47 } // namespace xla
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| sample_harness.cc | 19 #include "tensorflow/compiler/xla/array4d.h"
20 #include "tensorflow/compiler/xla/client/client.h"
21 #include "tensorflow/compiler/xla/client/client_library.h"
22 #include "tensorflow/compiler/xla/client/computation.h"
23 #include "tensorflow/compiler/xla/client/computation_builder.h"
24 #include "tensorflow/compiler/xla/client/global_data.h"
25 #include "tensorflow/compiler/xla/client/local_client.h"
26 #include "tensorflow/compiler/xla/literal_util.h"
27 #include "tensorflow/compiler/xla/statusor.h"
28 #include "tensorflow/compiler/xla/types.h
[all...] |
| cpu_copy_insertion.h | 19 #include "tensorflow/compiler/xla/service/hlo_module.h"
20 #include "tensorflow/compiler/xla/service/hlo_pass_interface.h"
22 namespace xla { namespace
24 // Besides the modifications made by the generic xla::CopyInsertion, this
40 } // namespace xla
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| interpreter_transfer_manager.h | 19 #include "tensorflow/compiler/xla/service/generic_transfer_manager.h"
22 namespace xla { namespace
24 // An implementation of the XLA GenericTransferManager for interpreter backend.
34 } // namespace xla
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| filecheck.h | 21 #include "tensorflow/compiler/xla/statusor.h"
22 #include "tensorflow/compiler/xla/types.h"
24 namespace xla { namespace
31 } // namespace xla
|
| local_client_aot_test_helper.cc | 16 // This program compiles an XLA program which computes 123 and writes the
23 #include "tensorflow/compiler/xla/client/client_library.h"
24 #include "tensorflow/compiler/xla/client/computation_builder.h"
25 #include "tensorflow/compiler/xla/service/cpu/cpu_compiler.h"
26 #include "tensorflow/compiler/xla/service/llvm_ir/llvm_util.h"
27 #include "tensorflow/compiler/xla/types.h"
28 #include "tensorflow/compiler/xla/util.h"
32 using xla::string;
34 xla::Computation Doubler(xla::Client* client)
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| triangular_solve_test.cc | 22 #include "tensorflow/compiler/xla/array2d.h"
23 #include "tensorflow/compiler/xla/client/computation_builder.h"
24 #include "tensorflow/compiler/xla/literal_util.h"
25 #include "tensorflow/compiler/xla/statusor.h"
26 #include "tensorflow/compiler/xla/test.h"
27 #include "tensorflow/compiler/xla/tests/client_library_test_base.h"
28 #include "tensorflow/compiler/xla/tests/literal_test_util.h"
29 #include "tensorflow/compiler/xla/tests/test_macros.h"
30 #include "tensorflow/compiler/xla/types.h"
36 using TriangularSolveTest = xla::ClientLibraryTestBase
[all...] |
| batch_dot.cc | 21 #include "tensorflow/compiler/xla/shape_util.h"
22 #include "tensorflow/compiler/xla/status_macros.h"
23 #include "tensorflow/compiler/xla/statusor.h"
28 xla::StatusOr<xla::ComputationDataHandle> BatchDot(
29 xla::ComputationBuilder* builder, xla::ComputationDataHandle x,
30 xla::ComputationDataHandle y, bool transpose_x, bool transpose_y,
32 TF_ASSIGN_OR_RETURN(std::unique_ptr<xla::Shape> x_shape,
34 TF_ASSIGN_OR_RETURN(std::unique_ptr<xla::Shape> y_shape
[all...] |
| /external/tensorflow/tensorflow/compiler/aot/ |
| compile.cc | 26 #include "tensorflow/compiler/xla/client/client_library.h"
27 #include "tensorflow/compiler/xla/client/compile_only_client.h"
28 #include "tensorflow/compiler/xla/service/cpu/cpu_compiler.h"
29 #include "tensorflow/compiler/xla/statusor.h"
30 #include "tensorflow/compiler/xla/util.h"
31 #include "tensorflow/compiler/xla/xla_data.pb.h"
44 // Compiles the XLA computation into executable code.
45 Status CompileXla(xla::CompileOnlyClient* client,
46 const xla::Computation& computation,
47 const xla::cpu::CpuAotCompilationOptions& aot_opts
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| flatten_call_graph.h | 21 #include "tensorflow/compiler/xla/service/hlo_pass_interface.h"
22 #include "tensorflow/compiler/xla/statusor.h"
24 namespace xla { namespace
38 } // namespace xla
|
| hlo_constant_folding.h | 19 #include "tensorflow/compiler/xla/service/hlo_module.h"
20 #include "tensorflow/compiler/xla/service/hlo_pass_interface.h"
22 namespace xla { namespace
35 } // namespace xla
|
| hlo_pass_interface.h | 19 #include "tensorflow/compiler/xla/service/hlo_module.h"
20 #include "tensorflow/compiler/xla/status_macros.h"
21 #include "tensorflow/compiler/xla/statusor.h"
22 #include "tensorflow/compiler/xla/types.h"
25 namespace xla { namespace
39 } // namespace xla
|
| hlo_profile_printer.h | 23 #include "tensorflow/compiler/xla/service/hlo_profile_printer_data.pb.h"
24 #include "tensorflow/compiler/xla/types.h"
26 namespace xla { namespace
30 } // namespace xla
|
| hlo_proto_util.h | 23 #include "tensorflow/compiler/xla/service/buffer_assignment.h"
24 #include "tensorflow/compiler/xla/service/hlo.pb.h"
25 #include "tensorflow/compiler/xla/service/hlo_module.h"
26 #include "tensorflow/compiler/xla/status.h"
28 namespace xla { namespace
38 } // namespace xla
|
| hlo_subcomputation_unification.h | 19 #include "tensorflow/compiler/xla/service/hlo_pass_interface.h"
21 namespace xla { namespace
34 } // namespace xla
|
| reshape_mover.h | 19 #include "tensorflow/compiler/xla/service/hlo_pass_interface.h"
21 namespace xla { namespace
34 } // namespace xla
|
| zero_sized_hlo_elimination.h | 19 #include "tensorflow/compiler/xla/service/hlo_module.h"
20 #include "tensorflow/compiler/xla/service/hlo_pass_interface.h"
23 namespace xla { namespace
31 } // namespace xla
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| while_transformer.h | 19 #include "tensorflow/compiler/xla/service/hlo_instruction.h"
20 #include "tensorflow/compiler/xla/statusor.h"
22 namespace xla { namespace
41 } // namespace xla
|
