Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DAdaptiveMaxPoolKernel.cpp162 Vec maxval_vec = Vec::loadu(out + d2); in cpu_adaptive_max_pool2d_channels_last() local
643 Vec maxval_vec = Vec::loadu(out + d2); in cpu_adaptive_max_pool3d_channels_last() local
H A DMaxPoolKernel.cpp113 Vec maxval_vec = Vec::loadu(out_data + d2); in compute_internal() local