| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | service_executable_run_options.h | 28 class ServiceExecutableRunOptions {
32 ServiceExecutableRunOptions() in ServiceExecutableRunOptions() function
33 : ServiceExecutableRunOptions(ExecutableRunOptions()) {} in ServiceExecutableRunOptions()
35 explicit ServiceExecutableRunOptions(ExecutableRunOptions run_options,
|
| D | executable.h | 259 const ServiceExecutableRunOptions* run_options,
282 const ServiceExecutableRunOptions* run_options,
289 const ServiceExecutableRunOptions* run_options,
294 const ServiceExecutableRunOptions* run_options,
303 absl::Span<const ServiceExecutableRunOptions> run_options,
320 const ServiceExecutableRunOptions* run_options,
324 const ServiceExecutableRunOptions* run_options,
328 const ServiceExecutableRunOptions* run_options,
332 const ServiceExecutableRunOptions* run_options,
|
| D | executable.cc | 82 const ServiceExecutableRunOptions* run_options, in ExecuteOnStream()
104 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
119 const ServiceExecutableRunOptions* run_options, in ExecuteOnStream()
131 absl::Span<const ServiceExecutableRunOptions> run_options, in ExecuteOnStreams()
163 const ServiceExecutableRunOptions* run_options, in ExecuteOnStreamWrapper()
174 const ServiceExecutableRunOptions* run_options, in ExecuteOnStreamWrapper()
192 const ServiceExecutableRunOptions* run_options) { in ExecuteWrapperBeforeExecution()
280 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStreamWrapper()
291 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStreamWrapper()
|
| D | hlo_runner.cc | 165 ServiceExecutableRunOptions service_run_options = in ExecuteWithDeviceBuffers()
194 const std::vector<ServiceExecutableRunOptions>&, in ExecuteReplicatedImpl()
202 std::vector<ServiceExecutableRunOptions> service_run_options; in ExecuteReplicatedImpl()
323 [&](const std::vector<ServiceExecutableRunOptions>& service_run_options, in ExecuteReplicated()
377 [&](const std::vector<ServiceExecutableRunOptions>& service_run_options, in ExecuteReplicated()
442 ServiceExecutableRunOptions HloRunner::GetServiceRunOptionsForDevice( in GetServiceRunOptionsForDevice()
455 return ServiceExecutableRunOptions(run_options, backend().StreamBorrower()); in GetServiceRunOptionsForDevice()
|
| D | hlo_runner.h | 148 ServiceExecutableRunOptions GetServiceRunOptionsForDevice(
155 const std::vector<ServiceExecutableRunOptions>&,
|
| D | service.cc | 409 ServiceExecutableRunOptions run_options(options, in ExecuteParallelAndRegisterResult()
492 std::vector<ServiceExecutableRunOptions> run_options; in ExecuteAndRegisterResult()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.h | 115 const ServiceExecutableRunOptions* run_options,
120 const ServiceExecutableRunOptions* run_options,
127 const ServiceExecutableRunOptions* run_options,
140 Status ExecuteThunks(const ServiceExecutableRunOptions* run_options,
157 const ServiceExecutableRunOptions* run_options);
|
| D | gpu_executable.cc | 95 const ServiceExecutableRunOptions* run_options) { in CheckCompatibilityWithServiceExecutableRunOptions()
129 const ServiceExecutableRunOptions* run_options, in ExecuteThunks()
399 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
407 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
417 const ServiceExecutableRunOptions* run_options, VariantArguments arguments, in ExecuteAsyncOnStreamImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executable_base.h | 41 const ServiceExecutableRunOptions* run_options,
47 const ServiceExecutableRunOptions* run_options,
|
| D | executable.h | 54 StatusOr<Literal> Evaluate(const ServiceExecutableRunOptions* run_options,
|
| D | executable.cc | 55 const ServiceExecutableRunOptions* run_options, in Evaluate()
|
| D | executable_base.cc | 42 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_executable_interface.h | 48 const ServiceExecutableRunOptions* run_options,
73 const ServiceExecutableRunOptions& run_options,
|
| D | tpu_executable.h | 53 const ServiceExecutableRunOptions& run_options,
|
| D | tpu_executable.cc | 38 const ServiceExecutableRunOptions& run_options, in LoadProgramAndEnqueueToStream()
|
| D | tpu_executable_interface.cc | 169 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_executable.h | 59 const ServiceExecutableRunOptions* run_options,
120 const ServiceExecutableRunOptions* run_options,
|
| D | cpu_executable.cc | 226 const ServiceExecutableRunOptions* run_options, in CreateResultShapedBuffer()
326 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
374 ServiceExecutableRunOptions run_options; in ExecuteAsyncOnStream()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_on_demand_compiler.cc | 41 const xla::ServiceExecutableRunOptions& options) { in ToC()
100 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
190 const ServiceExecutableRunOptions& run_options, in LoadProgramAndEnqueueToStream()
|
| D | tpu_execute.cc | 447 const xla::ServiceExecutableRunOptions service_run_options(run_options); in TPUExecute()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.h | 92 StatusOr<std::pair<ServiceExecutableRunOptions, StreamPool::Ptr>> RunHelper(
|
| D | local_client.cc | 118 StatusOr<std::pair<ServiceExecutableRunOptions, StreamPool::Ptr>>
167 ServiceExecutableRunOptions service_options(run_options, in RunHelper()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | mlir_gpu_test_base.cc | 89 ServiceExecutableRunOptions run_options(executable_run_options); in RunMlirModule()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | buffer_donation_test.cc | 87 ServiceExecutableRunOptions service_run_options(run_options); in RunAndCheck()
|
| D | xla_hlo_profile_test.cc | 175 ServiceExecutableRunOptions run_options(exec_run_options, in ExecuteAndFetchProfile()
|