| /external/tensorflow/tensorflow/compiler/xla/client/ |
| executable_build_options.cc | 23 ExecutableBuildOptions& ExecutableBuildOptions::set_device_allocator(
29 DeviceMemoryAllocator* ExecutableBuildOptions::device_allocator() const {
33 ExecutableBuildOptions& ExecutableBuildOptions::set_device_ordinal(
40 int ExecutableBuildOptions::device_ordinal() const { return device_ordinal_; }
42 ExecutableBuildOptions& ExecutableBuildOptions::set_result_layout(
49 const Shape* ExecutableBuildOptions::result_layout() const {
53 string ExecutableBuildOptions::ToString() const
[all...] |
| executable_build_options.h | 27 class ExecutableBuildOptions {
35 ExecutableBuildOptions& set_device_ordinal(int device_ordinal);
42 ExecutableBuildOptions& set_result_layout(const Shape& shape_with_layout);
52 ExecutableBuildOptions& set_device_allocator(
57 ExecutableBuildOptions& set_generate_hlo_graph(string regex);
|
| local_client.h | 55 const ExecutableBuildOptions& build_options() const { return build_options_; }
66 ExecutableBuildOptions build_options);
106 const ExecutableBuildOptions build_options_;
124 const ExecutableBuildOptions& options);
|
| local_client.cc | 45 ExecutableBuildOptions build_options)
252 const ExecutableBuildOptions& options) {
253 ExecutableBuildOptions updated_options = options;
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| local_service.h | 51 const ExecutableBuildOptions& options);
|
| local_service.cc | 75 const ExecutableBuildOptions& build_options) {
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| hlo_metadata_test.cc | 53 ExecutableBuildOptions()));
73 ExecutableBuildOptions());
|
| local_client_test_base.h | 104 const ExecutableBuildOptions& build_options,
113 const ExecutableBuildOptions& build_options,
117 ExecutableBuildOptions DefaultExecutableBuildOptions() const;
|
| local_client_test_base.cc | 144 ExecutableBuildOptions LocalClientTestBase::DefaultExecutableBuildOptions()
146 return ExecutableBuildOptions();
169 const ExecutableBuildOptions& build_options,
187 const ExecutableBuildOptions& build_options,
|
| local_client_allocation_test.cc | 92 computation, {}, ExecutableBuildOptions().set_device_ordinal(d),
|
| local_client_execute_test.cc | 272 ExecutableBuildOptions options = DefaultExecutableBuildOptions();
[all...] |
| xla_hlo_profile_test.cc | 147 ExecutableBuildOptions()));
|
| dynamic_ops_test.cc | 705 ExecutableBuildOptions())
|
| fusion_test.cc | [all...] |
| while_test.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| xla_jit_compiled_cpu_function.h | 47 const xla::ExecutableBuildOptions& build_options);
|
| xla_jit_compiled_cpu_function_test.cc | 80 xla::ExecutableBuildOptions()));
142 xla::ExecutableBuildOptions())
|
| xla_jit_compiled_cpu_function.cc | 114 const xla::ExecutableBuildOptions& build_options) {
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| dumped_computation_to_operation_list.cc | 88 ExecutableBuildOptions build_options;
|
| dumped_computation_to_text.cc | 64 ExecutableBuildOptions build_options;
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| local_computation_builder.cc | 259 const ExecutableBuildOptions* build_options) {
267 ExecutableBuildOptions options;
|
| local_computation_builder.h | 108 const ExecutableBuildOptions* build_options);
|
| local_computation_builder.i | 815 // ExecutableBuildOptions
817 %typemap(in) const ExecutableBuildOptions*
818 (ExecutableBuildOptions build_options) {
828 PyErr_SetString(PyExc_TypeError, "ExecutableBuildOptions.generate_hlo_graph must be a string or None.");
842 PyErr_SetString(PyExc_TypeError, tensorflow::strings::StrCat("ExecutableBuildOptions.result_shape could not be created from Python shape value: ", statusor.status().ToString()).c_str());
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| xla_compilation_cache.cc | 220 xla::ExecutableBuildOptions build_options;
|
