Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DCUDALoops.cuh251 void gpu_kernel_impl_nocast(TensorIteratorBase& iter, const func_t& f) { in gpu_kernel_impl_nocast() function
285 return gpu_kernel_impl_nocast(iter, f); in gpu_kernel_impl()
H A DLoops.cuh95 gpu_kernel_impl_nocast(iter, f); in gpu_kernel_nocast()