| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| executable.h | 21 #include "tensorflow/compiler/xla/service/executable.h"
22 #include "tensorflow/compiler/xla/service/hlo_cost_analysis.h"
23 #include "tensorflow/compiler/xla/service/hlo_execution_profile.h"
24 #include "tensorflow/compiler/xla/service/hlo_module.h"
25 #include "tensorflow/compiler/xla/service/hlo_module_config.h"
26 #include "tensorflow/compiler/xla/service/service_executable_run_options.h"
27 #include "tensorflow/compiler/xla/service/shaped_buffer.h"
28 #include "tensorflow/compiler/xla/statusor.h"
29 #include "tensorflow/compiler/xla/types.h"
30 #include "tensorflow/compiler/xla/xla_data.pb.h
36 namespace xla { namespace
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| liveness_util.h | 21 #include "tensorflow/compiler/xla/service/hlo_dataflow_analysis.h"
22 #include "tensorflow/compiler/xla/service/hlo_instruction.h"
23 #include "tensorflow/compiler/xla/service/tuple_points_to_analysis.h"
24 #include "tensorflow/compiler/xla/shape_util.h"
25 #include "tensorflow/compiler/xla/types.h"
27 namespace xla { namespace
62 } // namespace xla
|
| platform_util.h | 22 #include "tensorflow/compiler/xla/statusor.h"
23 #include "tensorflow/compiler/xla/types.h"
28 namespace xla { namespace
30 // Utilities for querying platforms and devices used by XLA.
33 // Returns the platforms present on the system and supported by XLA.
64 // element is nullptr, then the device is present by not supported by XLA.
74 } // namespace xla
|
| transpose_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
61 } // namespace xla
|
| versioned_computation_handle.h | 21 #include "tensorflow/compiler/xla/types.h"
22 #include "tensorflow/compiler/xla/xla_data.pb.h"
24 namespace xla { namespace
53 } // namespace xla
|
| tuple_util_test.cc | 16 #include "tensorflow/compiler/xla/service/tuple_util.h"
18 #include "tensorflow/compiler/xla/service/hlo_matchers.h"
19 #include "tensorflow/compiler/xla/test.h"
20 #include "tensorflow/compiler/xla/tools/parser/hlo_parser.h"
22 namespace xla { namespace
25 namespace op = ::xla::testing::opcode_matchers;
81 } // namespace xla
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| tuple_ops.h | 21 #include "tensorflow/compiler/xla/service/llvm_ir/ir_array.h"
27 namespace xla { namespace
81 } // namespace xla
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| shape_layout.h | 21 #include "tensorflow/compiler/xla/shape_util.h"
22 #include "tensorflow/compiler/xla/types.h"
23 #include "tensorflow/compiler/xla/xla_data.pb.h"
26 namespace xla { namespace
87 } // namespace xla
|
| text_literal_reader_test.cc | 16 #include "tensorflow/compiler/xla/text_literal_reader.h"
20 #include "tensorflow/compiler/xla/literal_util.h"
21 #include "tensorflow/compiler/xla/shape_util.h"
22 #include "tensorflow/compiler/xla/test.h"
23 #include "tensorflow/compiler/xla/types.h"
24 #include "tensorflow/compiler/xla/xla_data.pb.h"
27 namespace xla { namespace
58 } // namespace xla
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| deep_graph_test.cc | 16 #include "tensorflow/compiler/xla/tests/client_library_test_base.h"
18 namespace xla { namespace
38 } // namespace xla
|
| /external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
| hlo_token.h | 21 #include "tensorflow/compiler/xla/types.h"
24 namespace xla { namespace
76 } // namespace xla
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| xla_device_context.h | 21 #include "tensorflow/compiler/xla/client/global_data.h"
22 #include "tensorflow/compiler/xla/client/local_client.h"
29 // The allocator used for Tensors assigned to the XLA device. It uses
30 // XLA backend's allocator.
33 XlaDeviceAllocator(const xla::Backend* backend, int device_ordinal);
44 const xla::Backend* backend_;
49 // Helper class for managing data transfers between host and XLA devices.
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| fill_op.cc | 16 // XLA-specific Fill Op.
46 xla::Literal dims_literal;
59 xla::ComputationDataHandle data = ctx->Input(1);
|
| pack_op.cc | 16 // XLA Pack operator.
25 #include "tensorflow/compiler/xla/literal_util.h"
46 std::vector<xla::ComputationDataHandle> values;
72 std::vector<xla::ComputationDataHandle> reshaped_inputs(num);
|
| retval_op.cc | 19 #include "tensorflow/compiler/xla/client/computation_builder.h"
48 xla::ComputationDataHandle input = ctx->Input(0);
59 xla::Literal literal;
|
| sequence_ops.cc | 16 // XLA-specific sequence and range Ops.
21 #include "tensorflow/compiler/xla/literal_util.h"
33 xla::Literal literal;
40 xla::Literal literal;
43 case xla::S32:
46 case xla::S64:
58 Status CreateRangeTensor(const xla::Literal& start_literal,
59 const xla::Literal& limit_literal,
60 const xla::Literal& delta_literal, Tensor* output) {
111 xla::Literal start, limit, delta
[all...] |
| depthtospace_op.cc | 53 xla::ComputationBuilder* b = ctx->builder();
54 xla::ComputationDataHandle input = ctx->Input(0);
133 xla::ComputationDataHandle reshaped = b->Reshape(input, reshaped_shape);
144 xla::ComputationDataHandle permuted_reshaped =
155 xla::ComputationDataHandle output =
|
| spacetodepth_op.cc | 53 xla::ComputationBuilder* b = ctx->builder();
54 xla::ComputationDataHandle input = ctx->Input(0);
138 xla::ComputationDataHandle reshaped = b->Reshape(input, reshaped_shape);
148 xla::ComputationDataHandle permuted_reshaped =
159 xla::ComputationDataHandle output =
|
| concat_op.cc | 16 // XLA-specific Concat Ops.
25 #include "tensorflow/compiler/xla/literal_util.h"
52 xla::Literal literal;
57 std::vector<xla::ComputationDataHandle> values;
75 std::vector<xla::ComputationDataHandle> input_data;
79 xla::ComputationDataHandle handle = values[i];
163 xla::Literal inp0_literal;
167 xla::Literal concat_dim_literal;
183 xla::Literal inp_literal;
|
| fft_ops.cc | 16 // XLA-specific Ops for FFT.
21 #include "tensorflow/compiler/xla/literal_util.h"
37 using xla::FftType;
65 xla::ComputationBuilder* b = ctx->builder();
66 xla::ComputationDataHandle fft =
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| runtime_fork_join.cc | 16 #include "tensorflow/compiler/xla/service/cpu/runtime_fork_join.h"
21 #include "tensorflow/compiler/xla/executable_run_options.h"
70 const xla::ExecutableRunOptions* run_options =
71 static_cast<const xla::ExecutableRunOptions*>(run_options_ptr);
|
| /external/tensorflow/tensorflow/tools/ci_build/xla/linux/gpu/ |
| run_py3.sh | 42 --config=xla -- \
|
| /external/tensorflow/tensorflow/compiler/xla/legacy_flags/ |
| parse_flags_from_env_test.cc | 18 #include "tensorflow/compiler/xla/legacy_flags/parse_flags_from_env.h"
24 #include "tensorflow/compiler/xla/types.h"
31 namespace xla { namespace
34 // Test that XLA flags can be set from the environment.
162 } // namespace xla
166 xla::legacy_flags::binary_name = argv[0];
168 xla::int32 int_flag = 1;
174 xla::string usage = tensorflow::Flags::Usage(argv[0], flag_list);
175 bool parse_ok = xla::legacy_flags::ParseFlagsFromEnv(flag_list);
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| gpu_transfer_manager.cc | 16 #include "tensorflow/compiler/xla/service/gpu/gpu_transfer_manager.h"
23 #include "tensorflow/compiler/xla/literal_util.h"
24 #include "tensorflow/compiler/xla/service/gpu/gpu_compiler.h"
25 #include "tensorflow/compiler/xla/shape_util.h"
26 #include "tensorflow/compiler/xla/status_macros.h"
27 #include "tensorflow/compiler/xla/statusor.h"
28 #include "tensorflow/compiler/xla/types.h"
29 #include "tensorflow/compiler/xla/util.h"
30 #include "tensorflow/compiler/xla/xla_data.pb.h"
38 namespace xla { namespace
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| dumped_computation_to_text.cc | 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/local_client.h"
24 #include "tensorflow/compiler/xla/service/computation_tracker.h"
25 #include "tensorflow/compiler/xla/service/service.h"
26 #include "tensorflow/compiler/xla/service/session.pb.h"
27 #include "tensorflow/compiler/xla/statusor.h"
28 #include "tensorflow/compiler/xla/types.h"
29 #include "tensorflow/compiler/xla/xla_data.pb.h
35 namespace xla { namespace
[all...] |
