Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DAdaptiveMaxPoolKernel.cpp259 fVec max_fvec = fVec(-std::numeric_limits<float>::infinity()); in cpu_adaptive_max_pool2d_channels_last() local
747 fVec max_fvec = fVec(-std::numeric_limits<float>::infinity()); in cpu_adaptive_max_pool3d_channels_last() local
H A DSoftMaxKernel.cpp165 fVec max_fvec = fVec(-std::numeric_limits<float>::infinity()); in _vec_softmax_lastdim() local