Home
last modified time | relevance | path

Searched refs:parallel_reduce (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DReduceAllOpsKernel.cpp34 scalar_t result = at::parallel_reduce(0, input_numel, internal::GRAIN_SIZE, ident_v, in reduce_all_impl_vec()
54 scalar_t result = at::parallel_reduce(0, input_numel, internal::GRAIN_SIZE, ident_v, in reduce_all_impl()
127 scalar_t_pair result = at::parallel_reduce(0, input_numel, internal::GRAIN_SIZE, ident_v, in reduce_all_impl_two_outputs()
155 …std::pair<scalar_t, scalar_t> result = at::parallel_reduce(0, input_numel, internal::GRAIN_SIZE, i… in reduce_all_impl_vec_two_outputs()
H A DReduce.h265 iter.parallel_reduce([&](char** data, const int64_t* strides, int64_t size0, int64_t size1) {
H A DSumKernel.cpp544 iter.parallel_reduce( in cascade_sum()
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DParallel.h121 inline scalar_t parallel_reduce(
H A DParallel-inl.h46 inline scalar_t parallel_reduce( in parallel_reduce() function
H A DTensorIterator.h441 void parallel_reduce(loop2d_t loop);
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dtest_parallel.cpp84 at::parallel_reduce(0, 10, 1, 0, [&](int64_t begin, int64_t end, int ident) { in TEST()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorIteratorReduce.cpp24 void TensorIteratorBase::parallel_reduce(loop2d_t loop) { in parallel_reduce() function in at::TensorIteratorBase
H A DReduceOps.cpp1803 const double sum_dx2 = at::parallel_reduce( in std_var_all_cpu()
/aosp_15_r20/external/cronet/third_party/libc++/src/include/__pstl/internal/
H A Dparallel_backend_tbb.h119 return tbb::parallel_reduce( in __parallel_reduce()
202 …[__first, __last, &__body]() { tbb::parallel_reduce(tbb::blocked_range<_Index>(__first, __last, 3)… in __parallel_transform_reduce()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseCsrTensorMath.cpp1245 acc_t value = at::parallel_reduce( in reduce_sparse_csr_dim01_cpu_template()