| /external/tensorflow/tensorflow/compiler/xla/ |
| execution_options_util.cc | 21 ExecutionOptions execution_options; local
22 *(execution_options.mutable_debug_options()) =
24 return execution_options;
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| client.h | 45 // * If execution_options is not nullptr, these options are passed to the
48 // * If execution_options.device_handles is not empty, the computation is
57 const ExecutionOptions* execution_options = nullptr,
61 // * If execution_options.device_handles is not empty, the computation is
68 ExecutionOptions execution_options; member in struct:xla::Client::ComputationInstance
73 ExecutionOptions execution_options,
77 execution_options(execution_options),
137 const ExecutionOptions* execution_options = nullptr,
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| testing.cc | 73 auto execution_options = CreateDefaultExecutionOptions(); local
74 *execution_options.mutable_shape_with_output_layout() = shape;
75 return client->Execute(computation, /*arguments=*/{}, &execution_options)
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| compile_only_service.cc | 99 ExecutionOptions execution_options; local
100 *execution_options.mutable_debug_options() = debug_options;
104 &execution_options, *user_computation));
|
| local_service.cc | 121 ExecutionOptions execution_options = CreateDefaultExecutionOptions(); local
123 execution_options.mutable_debug_options()->set_xla_generate_hlo_graph(
127 *execution_options.mutable_shape_with_output_layout() =
130 *execution_options.mutable_shape_with_output_layout() =
133 execution_options.mutable_shape_with_output_layout());
137 CreateModuleConfig(*program_shape, argument_layouts, &execution_options,
|
| service.cc | 267 const ExecutionOptions* execution_options,
293 if (execution_options != nullptr &&
294 execution_options->has_shape_with_output_layout()) {
296 execution_options->shape_with_output_layout();
307 if (execution_options != nullptr) {
308 config->set_seed(execution_options->seed());
309 config->set_debug_options(execution_options->debug_options());
311 execution_options->debug_options().xla_hlo_profile());
327 const ExecutionOptions& execution_options,
333 return CreateModuleConfig(program_shape, argument_shapes, &execution_options,
715 const ExecutionOptions& execution_options = local
1241 ExecutionOptions execution_options = xla::CreateDefaultExecutionOptions(); local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| client_test.cc | 49 ExecutionOptions execution_options = execution_options_; local
50 *execution_options.mutable_shape_with_output_layout() =
55 client_->Execute(computation, {}, &execution_options));
79 ExecutionOptions execution_options = execution_options_; local
82 *execution_options.mutable_shape_with_output_layout() =
91 client_->ExecuteAndTransfer(computation, {}, &execution_options));
|
| reshape_test.cc | 649 ExecutionOptions execution_options = execution_options_; local
650 *execution_options.mutable_shape_with_output_layout() =
655 ->ExecuteAndTransfer(computation, {input.get()}, &execution_options)
808 ExecutionOptions execution_options = execution_options_; local
[all...] |
| prng_test.cc | 206 ExecutionOptions execution_options = execution_options_; local
207 execution_options.set_seed(125);
211 /*arguments=*/{param0_data.get()}, &execution_options));
|
| client_library_test_base.cc | 89 ExecutionOptions execution_options = execution_options_; local
91 *execution_options.mutable_shape_with_output_layout() =
95 &execution_options);
|
| while_test.cc | 953 ExecutionOptions execution_options = execution_options_; local
[all...] |
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| master_session.cc | 1051 GraphExecutionStateOptions execution_options; local
[all...] |
