Searched refs:ServiceExecutableRunOptions (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | executable.h | 83 const ServiceExecutableRunOptions* run_options,
90 const ServiceExecutableRunOptions* run_options,
103 const ServiceExecutableRunOptions* run_options, in ExecuteOnStream()
111 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
122 absl::Span<const ServiceExecutableRunOptions> run_options,
138 const ServiceExecutableRunOptions* run_options, ExecutionProfile* profile,
|
| 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.cc | 33 absl::Span<const ServiceExecutableRunOptions> run_options, in ExecuteOnStreams()
64 const ServiceExecutableRunOptions* run_options, ExecutionProfile* profile, in ExecuteOnStreamWrapper()
|
| D | hlo_runner.cc | 208 ServiceExecutableRunOptions service_run_options = in ExecuteWithDeviceBuffers()
245 ServiceExecutableRunOptions service_run_options = in ExecuteWithDeviceBuffers()
279 std::vector<ServiceExecutableRunOptions> service_run_options; in ExecuteReplicated()
442 ServiceExecutableRunOptions HloRunner::GetServiceRunOptionsForDevice( in GetServiceRunOptionsForDevice()
453 return ServiceExecutableRunOptions(run_options, backend().StreamBorrower()); in GetServiceRunOptionsForDevice()
|
| D | hlo_runner.h | 195 ServiceExecutableRunOptions GetServiceRunOptionsForDevice(
|
| D | service.cc | 454 ServiceExecutableRunOptions run_options(options, in ExecuteParallelAndRegisterResult()
540 std::vector<ServiceExecutableRunOptions> run_options; in ExecuteAndRegisterResult()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_executable.h | 59 const ServiceExecutableRunOptions* run_options,
64 const ServiceExecutableRunOptions* run_options,
95 const ServiceExecutableRunOptions* run_options,
128 const ServiceExecutableRunOptions* run_options,
|
| D | cpu_executable.cc | 209 const ServiceExecutableRunOptions* run_options, in CreateResultShapedBuffer()
268 const ServiceExecutableRunOptions* run_options, in ExecuteOnStream()
279 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
290 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStreamImpl()
327 ServiceExecutableRunOptions run_options; in ExecuteAsyncOnStreamImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.h | 80 const ServiceExecutableRunOptions* run_options,
85 const ServiceExecutableRunOptions* run_options,
94 Status ExecuteThunks(const ServiceExecutableRunOptions* run_options,
|
| D | gpu_executable.cc | 67 const ServiceExecutableRunOptions* run_options, in ExecuteThunks()
237 const ServiceExecutableRunOptions* run_options, in ExecuteOnStream()
350 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
|
| D | buffer_comparator.cc | 145 ServiceExecutableRunOptions service_run_options(run_options); in CompareEqualImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executable.h | 50 const ServiceExecutableRunOptions* run_options,
56 const ServiceExecutableRunOptions* run_options,
|
| D | executable.cc | 49 const ServiceExecutableRunOptions* run_options, in ExecuteOnStream()
125 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 168 ServiceExecutableRunOptions service_options(run_options, in Run()
179 const ServiceExecutableRunOptions* run_options, in ExecuteAndDump()
|
| D | local_client.h | 75 const ServiceExecutableRunOptions* run_options,
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | xla_hlo_profile_test.cc | 175 ServiceExecutableRunOptions run_options(exec_run_options, in ExecuteAndFetchProfile()
|