Searched refs:GetModuleFunction (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_driver.h | 232 static bool GetModuleFunction(GpuContext* context, GpuModuleHandle module,
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 309 if (!GpuDriver::GetModuleFunction(context_, module, kernelname->c_str(), in GetKernel()
1160 ptx_success = cuda::CUDADriver::GetModuleFunction( in PopulateDeviceDescription()
|
| D | cuda_driver.cc | 730 /* static */ bool GpuDriver::GetModuleFunction(GpuContext* context, in GetModuleFunction() function in stream_executor::gpu::GpuDriver
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 256 if (!GpuDriver::GetModuleFunction(context_, module, kernelname->c_str(), in GetKernel()
|
| D | rocm_driver.cc | 571 /* static */ bool GpuDriver::GetModuleFunction(GpuContext* context, in GetModuleFunction() function in stream_executor::gpu::GpuDriver
|