Searched defs:take_sqrt (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | ReduceMomentKernel.cu | 14 void std_var_kernel_impl(TensorIterator& iter, double correction, bool take_sqrt) { in std_var_kernel_impl()
23 static void std_var_kernel_cuda(TensorIterator& iter, double correction, bool take_sqrt) { in std_var_kernel_cuda()
|
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | ReduceOpsKernel.cpp | 138 static void std_var_kernel_impl(TensorIterator& iter, double correction, bool take_sqrt) { in std_var_kernel_impl()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | SharedReduceOps.h | 98 bool take_sqrt; member
|
| D | ReduceOps.cpp | 1769 static double std_var_all_cpu(const Tensor& self, double correction, bool take_sqrt) { in std_var_all_cpu()
1831 bool keepdim, bool take_sqrt) { in std_var_out()
1903 bool keepdim, bool take_sqrt) { in std_var_mean_out()
|