Searched defs:sum_fvec1 (Results 1 – 4 of 4) sorted by relevance
415 fVec sum_fvec1 = fVec::loadu(tmp_sum_data + d1 + fVec::size()); in _vec_softmax_backward() local450 fVec sum_fvec1 = fVec::loadu(tmp_sum_data + d2 + fVec::size()); in _vec_softmax_backward() local616 fVec sum_fvec1 = fVec::loadu(tmp_sum_data + d1 + fVec::size()); in _vec_log_softmax_backward() local649 fVec sum_fvec1 = fVec::loadu(tmp_sum_data + d2 + fVec::size()); in _vec_log_softmax_backward() local1081 fVec sum_fvec1 = fVec::loadu(tmp_sum_data + d2 + fVec::size()); in _vec_logsoftmax() local1112 fVec sum_fvec1 = fVec::loadu(tmp_sum_data + d3 + fVec::size()); in _vec_logsoftmax() local
224 fVec sum_fvec1 = fVec::loadu(sum + d2 + fVec::size()) + data_fvec1; in cpu_adaptive_avg_pool2d_channels_last() local648 fVec sum_fvec1 = fVec::loadu(sum + d2 + fVec::size()) + data_fvec1; in cpu_adaptive_avg_pool3d_channels_last() local
315 fVec sum_fvec1 = fVec::loadu(sum + d2 + fVec::size()) + data_fvec1; in cpu_avg_pool2d_channels_last() local874 fVec sum_fvec1 = fVec::loadu(sum + d2 + fVec::size()) + data_fvec1; in cpu_avg_pool3d_channels_last() local
930 fVec sum_fvec1 = fVec::loadu(buffer_ptr + d + fVec::size()) + data_fvec1; in batch_norm_cpu_collect_stats_channels_last_internal() local1184 fVec sum_fvec1 = dy_fvec1 + fVec::loadu(sum_ptr + d + fVec::size()); in batch_norm_cpu_backward_channels_last_internal() local