Searched refs:AsGpuKernel (Results 1 – 3 of 3) sorted by relevance
92 inline const GpuKernel* AsGpuKernel(const KernelBase* kernel) { in AsGpuKernel() function98 inline GpuKernel* AsGpuKernel(KernelBase* kernel) { in AsGpuKernel() function
265 GpuKernel* cuda_kernel = AsGpuKernel(kernel); in GetKernel()406 const GpuKernel* cuda_kernel = AsGpuKernel(&kernel); in Launch()460 const GpuKernel* cuda_kernel = AsGpuKernel(&kernel); in VlogOccupancyInfo()
240 GpuKernel* rocm_kernel = AsGpuKernel(kernel); in GetKernel()302 const GpuKernel* rocm_kernel = AsGpuKernel(&kernel); in Launch()