Home
last modified time | relevance | path

Searched refs:max_fvec1 (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DAdaptiveMaxPoolKernel.cpp293 fVec max_fvec1 = fVec::blendv(maxval_fvec1, val_fvec1, mask1); in cpu_adaptive_max_pool2d_channels_last() local
298 max_fvec1.store(max + d2 + fVec::size()); in cpu_adaptive_max_pool2d_channels_last()
318 fVec max_fvec1 = fVec::loadu(max + d3 + fVec::size()); in cpu_adaptive_max_pool2d_channels_last() local
319 bVec max_bvec = convert_from_float<scalar_t>(max_fvec0, max_fvec1); in cpu_adaptive_max_pool2d_channels_last()
782 fVec max_fvec1 = fVec::blendv(maxval_fvec1, val_fvec1, mask1); in cpu_adaptive_max_pool3d_channels_last() local
787 max_fvec1.store(max + d2 + fVec::size()); in cpu_adaptive_max_pool3d_channels_last()
808 fVec max_fvec1 = fVec::loadu(max + d3 + fVec::size()); in cpu_adaptive_max_pool3d_channels_last() local
809 bVec max_bvec = convert_float_bfloat16(max_fvec0, max_fvec1); in cpu_adaptive_max_pool3d_channels_last()
H A DSoftMaxKernel.cpp1054 fVec max_fvec1 = fVec::loadu(input_max_data + d1 + fVec::size()); in _vec_logsoftmax() local
1056 max_fvec1 = fVec::blendv(max_fvec1, data_fvec1, data_fvec1 > max_fvec1); in _vec_logsoftmax()
1083 fVec max_fvec1 = fVec::loadu(input_max_data + d2 + fVec::size()); in _vec_logsoftmax() local
1085 sum_fvec1 += (data_fvec1 - max_fvec1).exp(); in _vec_logsoftmax()
1110 fVec max_fvec1 = fVec::loadu(input_max_data + d3 + fVec::size()); in _vec_logsoftmax() local
1114 fVec out_fvec1 = data_fvec1 - max_fvec1 - sum_fvec1; in _vec_logsoftmax()
H A DMaxPoolKernel.cpp199 fVec max_fvec1 = fVec::blendv(maxval_fvec1, val_fvec1, mask1); in compute_internal() local
204 max_fvec1.store(max_ptr + d2 + fVec::size()); in compute_internal()
226 fVec max_fvec1 = fVec::loadu(max_ptr + d3 + fVec::size()); in compute_internal() local
227 Vec max_bvec = convert_from_float<scalar_t>(max_fvec0, max_fvec1); in compute_internal()