Home
last modified time | relevance | path

Searched defs:max_fvec0 (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DAdaptiveMaxPoolKernel.cpp292 fVec max_fvec0 = fVec::blendv(maxval_fvec0, val_fvec0, mask0); in cpu_adaptive_max_pool2d_channels_last() local
317 fVec max_fvec0 = fVec::loadu(max + d3); in cpu_adaptive_max_pool2d_channels_last() local
781 fVec max_fvec0 = fVec::blendv(maxval_fvec0, val_fvec0, mask0); in cpu_adaptive_max_pool3d_channels_last() local
807 fVec max_fvec0 = fVec::loadu(max + d3); in cpu_adaptive_max_pool3d_channels_last() local
H A DSoftMaxKernel.cpp1053 fVec max_fvec0 = fVec::loadu(input_max_data + d1); in _vec_logsoftmax() local
1082 fVec max_fvec0 = fVec::loadu(input_max_data + d2); in _vec_logsoftmax() local
1109 fVec max_fvec0 = fVec::loadu(input_max_data + d3); in _vec_logsoftmax() local
H A DMaxPoolKernel.cpp198 fVec max_fvec0 = fVec::blendv(maxval_fvec0, val_fvec0, mask0); in compute_internal() local
225 fVec max_fvec0 = fVec::loadu(max_ptr + d3); in compute_internal() local