Searched refs:GetModuleFunction (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/ |
| H A D | gpu_driver.h | 293 static bool GetModuleFunction(GpuContext* context, GpuModuleHandle module,
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_gpu_executor.cc | 271 if (!GpuDriver::GetModuleFunction(context_, module, kernelname->c_str(), in GetKernel()
|
| H A D | rocm_driver.cc | 520 /* static */ bool GpuDriver::GetModuleFunction(GpuContext* context, in GetModuleFunction() function in stream_executor::gpu::GpuDriver
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_gpu_executor.cc | 279 if (!GpuDriver::GetModuleFunction(context_, module, kernelname->c_str(), in GetKernel()
|
| H A D | cuda_driver.cc | 618 /* static */ bool GpuDriver::GetModuleFunction(GpuContext* context, in GetModuleFunction() function in stream_executor::gpu::GpuDriver
|