Searched refs:norm_kernel_cpu_impl (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | ReduceOpsKernel.cpp | 194 void norm_kernel_cpu_impl(TensorIterator& iter, const double& val) { in norm_kernel_cpu_impl() function
262 return norm_kernel_cpu_impl<at::Half, float>(iter, val); in norm_kernel_tensor_iterator_impl()
265 return norm_kernel_cpu_impl<at::Half, float, float>(iter, val); in norm_kernel_tensor_iterator_impl()
267 return norm_kernel_cpu_impl<at::BFloat16, float>(iter, val); in norm_kernel_tensor_iterator_impl()
270 return norm_kernel_cpu_impl<at::BFloat16, float, float>(iter, val); in norm_kernel_tensor_iterator_impl()
274 norm_kernel_cpu_impl<scalar_t>(iter, val); in norm_kernel_tensor_iterator_impl()
|