Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DIndexKernel.cpp70 constexpr int parallel_grain_size = 3000; in cpu_take_put_kernel() local
102 iter.for_each(loop, parallel_grain_size); in cpu_take_put_kernel()