| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | jitrt_custom_calls.cc | 68 const xla::ServiceExecutableRunOptions);
408 Error operator()(const ServiceExecutableRunOptions* run_options,
419 Error LaunchFunc::operator()(const ServiceExecutableRunOptions* run_options, in operator ()()
479 .UserData<const ServiceExecutableRunOptions*>() in LaunchFunc()
501 Error operator()(const ServiceExecutableRunOptions* run_options,
514 Error Gemm::operator()(const ServiceExecutableRunOptions* run_options, in operator ()()
550 .UserData<const ServiceExecutableRunOptions*>() in Gemm()
575 Error operator()(const ServiceExecutableRunOptions* run_options,
590 const ServiceExecutableRunOptions* run_options, in operator ()()
648 .UserData<const ServiceExecutableRunOptions*>() in CublasLtMatmul()
[all …]
|
| H A D | gpu_executable.h | 185 const ServiceExecutableRunOptions* run_options,
190 const ServiceExecutableRunOptions* run_options,
197 const ServiceExecutableRunOptions* run_options,
215 Status ExecuteThunksOrJitRt(const ServiceExecutableRunOptions* run_options,
238 const ServiceExecutableRunOptions* run_options);
|
| H A D | gpu_executable.cc | 285 const ServiceExecutableRunOptions* run_options) { in CheckCompatibilityWithServiceExecutableRunOptions()
315 Status MaybeSyncAndProfile(const ServiceExecutableRunOptions* run_options,
320 const ServiceExecutableRunOptions* run_options, in ExecuteThunks()
353 Status MaybeSyncAndProfile(const ServiceExecutableRunOptions* run_options, in MaybeSyncAndProfile()
556 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
563 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
574 const ServiceExecutableRunOptions* run_options, in ExecuteJitRt()
684 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStreamImpl()
845 const ServiceExecutableRunOptions* run_options, in ExecuteThunksOrJitRt()
|
| H A D | gpu_executable_run_options.cc | 54 const ServiceExecutableRunOptions& run_options, se::Stream* stream) in NcclExecuteParams()
|
| H A D | gpu_executable_run_options.h | 85 NcclExecuteParams(const ServiceExecutableRunOptions& run_options,
|
| H A D | thunk.h | 107 ExecuteParams(const ServiceExecutableRunOptions& run_options,
|
| H A D | thunk.cc | 22 const ServiceExecutableRunOptions& run_options, in ExecuteParams()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | executable.h | 257 const ServiceExecutableRunOptions* run_options,
280 const ServiceExecutableRunOptions* run_options,
287 const ServiceExecutableRunOptions* run_options,
292 const ServiceExecutableRunOptions* run_options,
301 absl::Span<const ServiceExecutableRunOptions> run_options,
318 const ServiceExecutableRunOptions* run_options,
322 const ServiceExecutableRunOptions* run_options,
326 const ServiceExecutableRunOptions* run_options,
330 const ServiceExecutableRunOptions* run_options,
|
| H A D | executable.cc | 81 const ServiceExecutableRunOptions* run_options, in ExecuteOnStream()
103 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
118 const ServiceExecutableRunOptions* run_options, in ExecuteOnStream()
130 absl::Span<const ServiceExecutableRunOptions> run_options, in ExecuteOnStreams()
162 const ServiceExecutableRunOptions* run_options, in ExecuteOnStreamWrapper()
173 const ServiceExecutableRunOptions* run_options, in ExecuteOnStreamWrapper()
191 const ServiceExecutableRunOptions* run_options) { in ExecuteWrapperBeforeExecution()
279 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStreamWrapper()
290 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStreamWrapper()
|
| H A 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,
|
| H A D | hlo_runner.cc | 176 ServiceExecutableRunOptions service_run_options = in ExecuteWithDeviceBuffers()
205 const std::vector<ServiceExecutableRunOptions>&, in ExecuteReplicatedImpl()
213 std::vector<ServiceExecutableRunOptions> service_run_options; in ExecuteReplicatedImpl()
345 [&](const std::vector<ServiceExecutableRunOptions>& service_run_options, in ExecuteReplicated()
405 [&](const std::vector<ServiceExecutableRunOptions>& service_run_options, in ExecuteReplicated()
472 ServiceExecutableRunOptions HloRunner::GetServiceRunOptionsForDevice( in GetServiceRunOptionsForDevice()
485 return ServiceExecutableRunOptions(run_options, backend().StreamBorrower()); in GetServiceRunOptionsForDevice()
|
| H A D | hlo_runner.h | 154 ServiceExecutableRunOptions GetServiceRunOptionsForDevice(
161 const std::vector<ServiceExecutableRunOptions>&,
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| H A D | executable_base.h | 41 const ServiceExecutableRunOptions* run_options,
47 const ServiceExecutableRunOptions* run_options,
|
| H A D | executable.cc | 55 const ServiceExecutableRunOptions* run_options, in Evaluate()
|
| H A D | executable.h | 51 StatusOr<Literal> Evaluate(const ServiceExecutableRunOptions* run_options,
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| H A D | tpu_executable_interface.h | 48 const ServiceExecutableRunOptions* run_options,
73 const ServiceExecutableRunOptions& run_options,
|
| H A D | tpu_executable.h | 34 const ServiceExecutableRunOptions* run_options,
49 const ServiceExecutableRunOptions& run_options, in LoadProgramAndEnqueueToStream()
|
| H A D | tpu_executable.cc | 26 const xla::ServiceExecutableRunOptions& options) { in ToC()
76 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
|
| H A D | tpu_op_executable.h | 53 const xla::ServiceExecutableRunOptions& run_options,
|
| H A D | tpu_op_executable.cc | 38 const xla::ServiceExecutableRunOptions& run_options, in LoadProgramAndEnqueueToStream()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_executable.h | 57 const ServiceExecutableRunOptions* run_options,
118 const ServiceExecutableRunOptions* run_options,
|
| H A D | cpu_executable.cc | 219 const ServiceExecutableRunOptions* run_options, in CreateResultShapedBuffer()
319 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream()
367 ServiceExecutableRunOptions run_options; in ExecuteAsyncOnStream()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | local_client.cc | 119 StatusOr<std::pair<ServiceExecutableRunOptions, StreamPool::Ptr>>
170 ServiceExecutableRunOptions service_options(run_options, in RunHelper()
|
| H A D | local_client.h | 93 StatusOr<std::pair<ServiceExecutableRunOptions, StreamPool::Ptr>> RunHelper(
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/tpu/ |
| H A D | tpu_op_executable.cc | 38 const xla::ServiceExecutableRunOptions& run_options, in LoadProgramAndEnqueueToStream()
|