| /external/tensorflow/tensorflow/compiler/xla/ |
| execution_options_util.cc | 21 ExecutionOptions execution_options; local
22 *(execution_options.mutable_debug_options()) = GetDebugOptionsFromFlags();
23 return execution_options;
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| client.h | 50 // * If execution_options is not nullptr, these options are passed to the
53 // * If execution_options.device_handles should be empty. If you need
64 const ExecutionOptions* execution_options = nullptr);
76 // * If execution_options is not nullptr, these options are passed to the
79 // * If execution_options.device_handles is not empty, the computation is
92 const ExecutionOptions* execution_options = nullptr,
96 // * If execution_options.device_handles is not empty, the computation is
103 ExecutionOptions execution_options; member in struct:xla::Client::XlaComputationInstance
108 ExecutionOptions execution_options,
112 execution_options(execution_options)
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| compile_only_service.cc | 74 ExecutionOptions execution_options; local
75 *execution_options.mutable_debug_options() = debug_options;
76 *execution_options.mutable_shape_with_output_layout() =
80 execution_options.mutable_device_assignment()));
86 instance.argument_layouts, &execution_options));
|
| local_service.cc | 100 ExecutionOptions execution_options = CreateDefaultExecutionOptions(); local
102 *execution_options.mutable_debug_options() = build_options.debug_options();
105 *execution_options.mutable_shape_with_output_layout() =
110 *execution_options.mutable_shape_with_output_layout() =
113 execution_options.set_num_replicas(build_options.num_replicas());
114 return execution_options;
164 ExecutionOptions execution_options = local
169 CreateModuleConfig(program_shape, argument_layouts, &execution_options));
|
| service.cc | 267 const ExecutionOptions* execution_options) {
291 if (execution_options != nullptr &&
292 execution_options->has_shape_with_output_layout()) {
294 execution_options->shape_with_output_layout());
305 if (execution_options != nullptr) {
306 if (execution_options->num_replicas() > 0) {
307 config->set_replica_count(execution_options->num_replicas());
311 config->set_seed(execution_options->seed());
312 config->set_debug_options(execution_options->debug_options());
324 if (execution_options != nullptr &
640 const ExecutionOptions& execution_options = local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| testing.cc | 68 auto execution_options = CreateDefaultExecutionOptions(); local
69 *execution_options.mutable_shape_with_output_layout() = shape.ToProto();
71 *execution_options.mutable_debug_options() = *debug_opts;
73 return client->Execute(computation, /*arguments=*/{}, &execution_options)
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| client_test.cc | 50 ExecutionOptions execution_options = execution_options_; local
51 *execution_options.mutable_shape_with_output_layout() =
57 client_->Execute(computation, {}, &execution_options));
80 ExecutionOptions execution_options = execution_options_; local
83 *execution_options.mutable_shape_with_output_layout() =
93 client_->ExecuteAndTransfer(computation, {}, &execution_options));
|
| gather_operation_test.cc | 663 xla::ExecutionOptions execution_options = CreateDefaultExecutionOptions(); local
664 *execution_options.add_device_handles() = devices[0];
669 execution_options,
|
| reshape_test.cc | 618 ExecutionOptions execution_options = execution_options_; local
619 *execution_options.mutable_shape_with_output_layout() =
625 ->ExecuteAndTransfer(computation, {input.get()}, &execution_options)
768 ExecutionOptions execution_options = execution_options_; local
[all...] |
| prng_test.cc | 207 ExecutionOptions execution_options = execution_options_; local
208 execution_options.set_seed(125);
212 /*arguments=*/{param0_data.get()}, &execution_options));
299 ExecutionOptions execution_options = execution_options_; local
300 execution_options.set_seed(42);
307 &execution_options));
|
| client_library_test_base.cc | 115 ExecutionOptions execution_options = execution_options_; local
117 *execution_options.mutable_shape_with_output_layout() =
121 &execution_options);
135 ExecutionOptions execution_options = execution_options_; local
137 *execution_options.mutable_shape_with_output_layout() =
140 execution_options.clear_device_handles();
142 &execution_options);
|
| while_test.cc | 942 ExecutionOptions execution_options = execution_options_; local
[all...] |
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| master_session.cc | 1221 GraphExecutionStateOptions execution_options; local
[all...] |
