| /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 | 171 ServiceExecutableRunOptions service_run_options = in ExecuteWithDeviceBuffers()
200 const std::vector<ServiceExecutableRunOptions>&, in ExecuteReplicatedImpl()
208 std::vector<ServiceExecutableRunOptions> service_run_options; in ExecuteReplicatedImpl()
339 [&](const std::vector<ServiceExecutableRunOptions>& service_run_options, in ExecuteReplicated()
393 [&](const std::vector<ServiceExecutableRunOptions>& service_run_options, in ExecuteReplicated()
458 ServiceExecutableRunOptions HloRunner::GetServiceRunOptionsForDevice( in GetServiceRunOptionsForDevice()
471 return ServiceExecutableRunOptions(run_options, backend().StreamBorrower()); in GetServiceRunOptionsForDevice()
|
| D | hlo_runner.h | 147 ServiceExecutableRunOptions GetServiceRunOptionsForDevice(
154 const std::vector<ServiceExecutableRunOptions>&,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.h | 121 const ServiceExecutableRunOptions* run_options,
126 const ServiceExecutableRunOptions* run_options,
133 const ServiceExecutableRunOptions* run_options,
147 const ServiceExecutableRunOptions* run_options,
163 const ServiceExecutableRunOptions* run_options);
|
| D | gpu_executable.cc | 120 const ServiceExecutableRunOptions* run_options) { in CheckCompatibilityWithServiceExecutableRunOptions()
150 const ServiceExecutableRunOptions* run_options, in ExecuteThunks()
413 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
420 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
521 const ServiceExecutableRunOptions* run_options, in ExecuteBef()
599 const ServiceExecutableRunOptions* run_options, 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 | 34 const ServiceExecutableRunOptions* run_options,
49 const ServiceExecutableRunOptions& run_options, in LoadProgramAndEnqueueToStream()
|
| D | tpu_op_executable.h | 53 const xla::ServiceExecutableRunOptions& run_options,
|
| D | tpu_executable.cc | 25 const xla::ServiceExecutableRunOptions& options) { in ToC()
75 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
|
| D | tpu_op_executable.cc | 38 const xla::ServiceExecutableRunOptions& run_options, in LoadProgramAndEnqueueToStream()
|
| D | tpu_executable_interface.cc | 184 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 | 229 const ServiceExecutableRunOptions* run_options, in CreateResultShapedBuffer()
329 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
377 ServiceExecutableRunOptions run_options; in ExecuteAsyncOnStream()
|
| /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>>
169 ServiceExecutableRunOptions service_options(run_options, in RunHelper()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | mlir_gpu_test_base.cc | 79 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()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.cc | 447 const xla::ServiceExecutableRunOptions service_run_options(run_options); in TPUExecute()
|