Searched refs:norm_kernel_cuda_impl (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | ReduceNormKernel.cu | 17 void norm_kernel_cuda_impl(TensorIterator& iter, double p) { in norm_kernel_cuda_impl() function
35 return norm_kernel_cuda_impl<at::Half, float>(iter, ord); in norm_launch_kernel()
38 return norm_kernel_cuda_impl<at::Half, float, float>(iter, ord); in norm_launch_kernel()
41 return norm_kernel_cuda_impl<at::BFloat16, float>(iter, ord); in norm_launch_kernel()
44 return norm_kernel_cuda_impl<at::BFloat16, float, float>(iter, ord); in norm_launch_kernel()
47 norm_kernel_cuda_impl<scalar_t>(iter, ord); in norm_launch_kernel()
|