Searched defs:max_fvec1 (Results 1 – 3 of 3) sorted by relevance
293 fVec max_fvec1 = fVec::blendv(maxval_fvec1, val_fvec1, mask1); in cpu_adaptive_max_pool2d_channels_last() local318 fVec max_fvec1 = fVec::loadu(max + d3 + fVec::size()); in cpu_adaptive_max_pool2d_channels_last() local782 fVec max_fvec1 = fVec::blendv(maxval_fvec1, val_fvec1, mask1); in cpu_adaptive_max_pool3d_channels_last() local808 fVec max_fvec1 = fVec::loadu(max + d3 + fVec::size()); in cpu_adaptive_max_pool3d_channels_last() local
1054 fVec max_fvec1 = fVec::loadu(input_max_data + d1 + fVec::size()); in _vec_logsoftmax() local1083 fVec max_fvec1 = fVec::loadu(input_max_data + d2 + fVec::size()); in _vec_logsoftmax() local1110 fVec max_fvec1 = fVec::loadu(input_max_data + d3 + fVec::size()); in _vec_logsoftmax() local
199 fVec max_fvec1 = fVec::blendv(maxval_fvec1, val_fvec1, mask1); in compute_internal() local226 fVec max_fvec1 = fVec::loadu(max_ptr + d3 + fVec::size()); in compute_internal() local