Searched refs:mean_kernel_impl (Results 1 – 1 of 1) sorted by relevance
40 void mean_kernel_impl(TensorIterator& iter) { in mean_kernel_impl() function49 mean_kernel_impl<at::Half, float>(iter); in mean_kernel_cuda()52 mean_kernel_impl<at::Half, float, float>(iter); in mean_kernel_cuda()54 mean_kernel_impl<at::BFloat16, float>(iter); in mean_kernel_cuda()57 mean_kernel_impl<at::BFloat16, float, float>(iter); in mean_kernel_cuda()60 mean_kernel_impl<scalar_t>(iter); in mean_kernel_cuda()