Home
last modified time | relevance | path

Searched refs:cufunc (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc426 CUfunction cufunc = cuda_kernel->AsGpuFunctionHandle(); in Launch() local
433 if (!launched_kernels_.count(cufunc)) { in Launch()
437 launched_kernels_.insert(cufunc); in Launch()
443 GpuDriver::FuncSetCacheConfig(cufunc, cuda_kernel->GetGpuCacheConfig()); in Launch()
448 if (!GpuDriver::LaunchKernel(context_, cufunc, block_dims.x, block_dims.y, in Launch()
488 CUfunction cufunc = cuda_kernel->AsGpuFunctionHandle(); in VlogOccupancyInfo() local
491 smem_per_block, thread_dims, cufunc); in VlogOccupancyInfo()
496 smem_per_block, thread_dims, cufunc); in VlogOccupancyInfo()