Searched refs:binary_kernel_reduce (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | ReduceOpsKernel.cpp | 140 binary_kernel_reduce( in std_var_kernel_impl()
196 binary_kernel_reduce(iter, NormZeroOps<scalar_t, acc_t, out_t>(), acc_t(0)); in norm_kernel_cpu_impl()
198 binary_kernel_reduce(iter, NormOneOps<scalar_t, acc_t, out_t>(), acc_t(0)); in norm_kernel_cpu_impl()
200 binary_kernel_reduce(iter, NormTwoOps<scalar_t, acc_t, out_t>(), acc_t(0)); in norm_kernel_cpu_impl()
202 binary_kernel_reduce(iter, AbsMaxOps<scalar_t, acc_t, out_t>(), acc_t(0)); in norm_kernel_cpu_impl()
204 …binary_kernel_reduce(iter, AbsMinOps<scalar_t, acc_t, out_t>(), std::numeric_limits<acc_t>::infini… in norm_kernel_cpu_impl()
206 binary_kernel_reduce(iter, NormOps<scalar_t, acc_t, out_t>{acc_t(val)}, acc_t(0)); in norm_kernel_cpu_impl()
363 binary_kernel_reduce( in min_values_kernel_impl()
405 binary_kernel_reduce( in argmax_kernel_impl()
429 binary_kernel_reduce( in argmin_kernel_impl()
|
| H A D | Reduce.h | 189 void binary_kernel_reduce(TensorIteratorBase& iter, ops_t ops, init_t init) { in binary_kernel_reduce() function
|