Searched refs:GpuKernel (Results 1 – 7 of 7) sorted by relevance
36 class GpuKernel : public internal::KernelInterface {38 GpuKernel() in GpuKernel() function45 ~GpuKernel() override {} in ~GpuKernel()92 inline const GpuKernel* AsGpuKernel(const KernelBase* kernel) { in AsGpuKernel()93 return static_cast<const GpuKernel*>(kernel->implementation()); in AsGpuKernel()98 inline GpuKernel* AsGpuKernel(KernelBase* kernel) { in AsGpuKernel()99 return static_cast<GpuKernel*>(kernel->implementation()); in AsGpuKernel()
262 port::Status GetKernelMetadata(GpuKernel* cuda_kernel,
30 using CUDAKernel = gpu::GpuKernel;
21 CUfunc_cache GpuKernel::GetGpuCacheConfig() const { in GetGpuCacheConfig()
265 GpuKernel* cuda_kernel = AsGpuKernel(kernel); in GetKernel()386 port::Status GpuExecutor::GetKernelMetadata(GpuKernel* cuda_kernel, in GetKernelMetadata()406 const GpuKernel* cuda_kernel = AsGpuKernel(&kernel); in Launch()460 const GpuKernel* cuda_kernel = AsGpuKernel(&kernel); in VlogOccupancyInfo()877 return std::unique_ptr<internal::KernelInterface>(new GpuKernel()); in CreateKernelImplementation()
21 hipFuncCache_t GpuKernel::GetGpuCacheConfig() const { in GetGpuCacheConfig()
240 GpuKernel* rocm_kernel = AsGpuKernel(kernel); in GetKernel()285 port::Status GpuExecutor::GetKernelMetadata(GpuKernel* rocm_kernel, in GetKernelMetadata()302 const GpuKernel* rocm_kernel = AsGpuKernel(&kernel); in Launch()782 return std::unique_ptr<internal::KernelInterface>(new GpuKernel()); in CreateKernelImplementation()