| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | client.cc | 168 const ExecutionOptions* execution_options, in ExecuteAndTransfer() argument
172 Execute(computation, arguments, execution_options, execution_profile)); in ExecuteAndTransfer()
175 if (execution_options && execution_options->has_shape_with_output_layout()) { in ExecuteAndTransfer()
177 Shape(execution_options->shape_with_output_layout()); in ExecuteAndTransfer()
219 const ExecutionOptions* execution_options) { in Compile() argument
223 if (execution_options == nullptr) { in Compile()
226 *request.mutable_execution_options() = *execution_options; in Compile()
228 if (request.execution_options().device_handles_size() > 1) { in Compile()
279 const ExecutionOptions* execution_options, in Execute() argument
283 if (!execution_options || execution_options->device_handles().empty()) { in Execute()
[all …]
|
| D | client.h | 64 const ExecutionOptions* execution_options = nullptr);
92 const ExecutionOptions* execution_options = nullptr,
103 ExecutionOptions execution_options; member
108 ExecutionOptions execution_options, in XlaComputationInstance()
112 execution_options(execution_options), in XlaComputationInstance()
173 const ExecutionOptions* execution_options = nullptr,
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | local_service.cc | 100 ExecutionOptions execution_options = CreateDefaultExecutionOptions(); in CreateExecutionOptions() local
102 *execution_options.mutable_debug_options() = build_options.debug_options(); in CreateExecutionOptions()
105 *execution_options.mutable_shape_with_output_layout() = in CreateExecutionOptions()
110 *execution_options.mutable_shape_with_output_layout() = in CreateExecutionOptions()
113 execution_options.set_num_replicas(build_options.num_replicas()); in CreateExecutionOptions()
114 return execution_options; in CreateExecutionOptions()
164 ExecutionOptions execution_options = in CompileExecutable() local
169 CreateModuleConfig(program_shape, argument_layouts, &execution_options)); in CompileExecutable()
|
| D | service.cc | 267 const ExecutionOptions* execution_options) { in CreateModuleConfig() argument
291 if (execution_options != nullptr && in CreateModuleConfig()
292 execution_options->has_shape_with_output_layout()) { in CreateModuleConfig()
294 execution_options->shape_with_output_layout()); in CreateModuleConfig()
305 if (execution_options != nullptr) { in CreateModuleConfig()
306 if (execution_options->num_replicas() > 0) { in CreateModuleConfig()
307 config->set_replica_count(execution_options->num_replicas()); in CreateModuleConfig()
311 config->set_seed(execution_options->seed()); in CreateModuleConfig()
312 config->set_debug_options(execution_options->debug_options()); in CreateModuleConfig()
324 if (execution_options != nullptr && in CreateModuleConfig()
[all …]
|
| D | compile_only_service.cc | 74 ExecutionOptions execution_options; in CompileAheadOfTime() local
75 *execution_options.mutable_debug_options() = debug_options; in CompileAheadOfTime()
76 *execution_options.mutable_shape_with_output_layout() = in CompileAheadOfTime()
80 execution_options.mutable_device_assignment())); in CompileAheadOfTime()
86 instance.argument_layouts, &execution_options)); in CompileAheadOfTime()
|
| D | service.h | 192 const ExecutionOptions& execution_options);
196 const ExecutionOptions& execution_options, int64 requests_size,
201 const ExecutionOptions& execution_options,
226 const ExecutionOptions* execution_options);
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | client_test.cc | 50 ExecutionOptions execution_options = execution_options_; in XLA_TEST_F() local
51 *execution_options.mutable_shape_with_output_layout() = in XLA_TEST_F()
57 client_->Execute(computation, {}, &execution_options)); in XLA_TEST_F()
80 ExecutionOptions execution_options = execution_options_; in XLA_TEST_F() local
83 *execution_options.mutable_shape_with_output_layout() = in XLA_TEST_F()
93 client_->ExecuteAndTransfer(computation, {}, &execution_options)); in XLA_TEST_F()
|
| D | prng_test.cc | 207 ExecutionOptions execution_options = execution_options_; in XLA_TEST_F() local
208 execution_options.set_seed(125); in XLA_TEST_F()
212 /*arguments=*/{param0_data.get()}, &execution_options)); in XLA_TEST_F()
299 ExecutionOptions execution_options = execution_options_; in XLA_TEST_F() local
300 execution_options.set_seed(42); in XLA_TEST_F()
307 &execution_options)); in XLA_TEST_F()
|
| D | client_library_test_base.cc | 115 ExecutionOptions execution_options = execution_options_; in ExecuteAndTransfer() local
117 *execution_options.mutable_shape_with_output_layout() = in ExecuteAndTransfer()
121 &execution_options); in ExecuteAndTransfer()
135 ExecutionOptions execution_options = execution_options_; in ExecuteAndTransferReference() local
137 *execution_options.mutable_shape_with_output_layout() = in ExecuteAndTransferReference()
140 execution_options.clear_device_handles(); in ExecuteAndTransferReference()
142 &execution_options); in ExecuteAndTransferReference()
|
| D | gather_operation_test.cc | 663 xla::ExecutionOptions execution_options = CreateDefaultExecutionOptions(); in XLA_TEST_F() local
664 *execution_options.add_device_handles() = devices[0]; in XLA_TEST_F()
669 execution_options, in XLA_TEST_F()
|
| D | reshape_test.cc | 618 ExecutionOptions execution_options = execution_options_; in XLA_TEST_P() local
619 *execution_options.mutable_shape_with_output_layout() = in XLA_TEST_P()
625 ->ExecuteAndTransfer(computation, {input.get()}, &execution_options) in XLA_TEST_P()
768 ExecutionOptions execution_options = execution_options_; in XLA_TEST_P() local
769 *execution_options.mutable_shape_with_output_layout() = in XLA_TEST_P()
776 &execution_options) in XLA_TEST_P()
|
| D | while_test.cc | 942 ExecutionOptions execution_options = execution_options_; in XLA_TEST_F() local
943 execution_options.set_seed(65); in XLA_TEST_F()
946 client_->ExecuteAndTransfer(computation, {}, &execution_options)); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | execution_options_util.cc | 21 ExecutionOptions execution_options; in CreateDefaultExecutionOptions() local
22 *(execution_options.mutable_debug_options()) = GetDebugOptionsFromFlags(); in CreateDefaultExecutionOptions()
23 return execution_options; in CreateDefaultExecutionOptions()
|
| D | xla.proto | 392 ExecutionOptions execution_options = 2; field
420 ExecutionOptions execution_options = 3; field
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | testing.cc | 68 auto execution_options = CreateDefaultExecutionOptions(); in MakeFakeDataViaDeviceOrDie() local
69 *execution_options.mutable_shape_with_output_layout() = shape.ToProto(); in MakeFakeDataViaDeviceOrDie()
71 *execution_options.mutable_debug_options() = *debug_opts; in MakeFakeDataViaDeviceOrDie()
73 return client->Execute(computation, /*arguments=*/{}, &execution_options) in MakeFakeDataViaDeviceOrDie()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | master_session.cc | 1221 GraphExecutionStateOptions execution_options; in Create() local
1222 execution_options.device_set = devices_.get(); in Create()
1223 execution_options.session_options = &session_opts_; in Create()
1227 graph_def, execution_options, &execution_state_)); in Create()
|