Home
last modified time | relevance | path

Searched defs:sum_arr (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DAdaptiveAvgPoolKernel.cpp188 auto sum_arr = std::make_unique<float []>(channels); in cpu_adaptive_avg_pool2d_channels_last() local
606 auto sum_arr = std::make_unique<float []>(channels); in cpu_adaptive_avg_pool3d_channels_last() local
H A DAvgPoolKernel.cpp253 auto sum_arr = std::make_unique<float []>(channels); in cpu_avg_pool2d_channels_last() local
807 auto sum_arr = std::make_unique<float []>(channels); in cpu_avg_pool3d_channels_last() local