Searched refs:std_var_kernel_impl (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | ReduceMomentKernel.cu | 14 void std_var_kernel_impl(TensorIterator& iter, double correction, bool take_sqrt) { in std_var_kernel_impl() function
27 std_var_kernel_impl<at::Half, float>(iter, correction, take_sqrt); in std_var_kernel_cuda()
30 std_var_kernel_impl<at::BFloat16, float>(iter, correction, take_sqrt); in std_var_kernel_cuda()
34 std_var_kernel_impl<scalar_t>(iter, correction, take_sqrt); in std_var_kernel_cuda()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | ReduceOpsKernel.cpp | 138 static void std_var_kernel_impl(TensorIterator& iter, double correction, bool take_sqrt) { in std_var_kernel_impl() function
438 REGISTER_DISPATCH(std_var_stub, &std_var_kernel_impl);
|