Searched refs:binary_kernel_reduce_lastdim (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | ReduceOpsKernel.cpp | 236 …binary_kernel_reduce_lastdim(iter, [](char* result_data_bytes, char* self_data_bytes, int64_t size… in norm_kernel_tensor_iterator_impl()
393 …binary_kernel_reduce_lastdim(iter, [&](char* result_data_bytes, char* self_data_bytes, int64_t siz… in argmax_kernel_impl()
417 …binary_kernel_reduce_lastdim(iter, [&](char* result_data_bytes, char* self_data_bytes, int64_t siz… in argmin_kernel_impl()
|
| H A D | Reduce.h | 295 void binary_kernel_reduce_lastdim(TensorIteratorBase& iter, reduce_func_t reduce_op) { in binary_kernel_reduce_lastdim() function
|