Searched refs:cufunc (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 407 CUfunction cufunc = cuda_kernel->AsGpuFunctionHandle(); in Launch() local
414 if (!launched_kernels_.count(cufunc)) { in Launch()
418 launched_kernels_.insert(cufunc); in Launch()
425 cufunc, cuda_kernel->GetGpuCacheConfig())); in Launch()
431 context_, cufunc, block_dims.x, block_dims.y, block_dims.z, thread_dims.x, in Launch()
461 CUfunction cufunc = cuda_kernel->AsGpuFunctionHandle(); in VlogOccupancyInfo() local
464 smem_per_block, thread_dims, cufunc); in VlogOccupancyInfo()
469 smem_per_block, thread_dims, cufunc); in VlogOccupancyInfo()
|