Home
last modified time | relevance | path

Searched defs:sum_fvec0 (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DSoftMaxKernel.cpp414 fVec sum_fvec0 = fVec::loadu(tmp_sum_data + d1); in _vec_softmax_backward() local
449 fVec sum_fvec0 = fVec::loadu(tmp_sum_data + d2); in _vec_softmax_backward() local
615 fVec sum_fvec0 = fVec::loadu(tmp_sum_data + d1); in _vec_log_softmax_backward() local
648 fVec sum_fvec0 = fVec::loadu(tmp_sum_data + d2); in _vec_log_softmax_backward() local
1080 fVec sum_fvec0 = fVec::loadu(tmp_sum_data + d2); in _vec_logsoftmax() local
1111 fVec sum_fvec0 = fVec::loadu(tmp_sum_data + d3); in _vec_logsoftmax() local
H A DAdaptiveAvgPoolKernel.cpp223 fVec sum_fvec0 = fVec::loadu(sum + d2) + data_fvec0; in cpu_adaptive_avg_pool2d_channels_last() local
647 fVec sum_fvec0 = fVec::loadu(sum + d2) + data_fvec0; in cpu_adaptive_avg_pool3d_channels_last() local
H A DAvgPoolKernel.cpp314 fVec sum_fvec0 = fVec::loadu(sum + d2) + data_fvec0; in cpu_avg_pool2d_channels_last() local
873 fVec sum_fvec0 = fVec::loadu(sum + d2) + data_fvec0; in cpu_avg_pool3d_channels_last() local
H A Dbatch_norm_kernel.cpp929 fVec sum_fvec0 = fVec::loadu(buffer_ptr + d) + data_fvec0; in batch_norm_cpu_collect_stats_channels_last_internal() local
1183 fVec sum_fvec0 = dy_fvec0 + fVec::loadu(sum_ptr + d); in batch_norm_cpu_backward_channels_last_internal() local