Home
last modified time | relevance | path

Searched defs:cuda_kernel (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_gpu_executor.cc245 GpuKernel* cuda_kernel = AsGpuKernel(kernel); in GetKernel() local
427 port::Status GpuExecutor::GetKernelMetadata(GpuKernel* cuda_kernel, in GetKernelMetadata()
447 const GpuKernel* cuda_kernel = AsGpuKernel(&kernel); in Launch() local
502 const GpuKernel* cuda_kernel = AsGpuKernel(&kernel); in VlogOccupancyInfo() local
/aosp_15_r20/external/pytorch/torch/utils/hipify/
H A Dhipify_python.py231 def add_dim3(kernel_string, cuda_kernel): argument