Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DAdaptiveAvgPoolKernel.cpp206 fVec sum_fvec = fVec(float(0)); in cpu_adaptive_avg_pool2d_channels_last() local
628 fVec sum_fvec = fVec(float(0)); in cpu_adaptive_avg_pool3d_channels_last() local
H A DAvgPoolKernel.cpp285 fVec sum_fvec = fVec(float(0)); in cpu_avg_pool2d_channels_last() local
843 fVec sum_fvec = fVec(float(0)); in cpu_avg_pool3d_channels_last() local
H A Dbatch_norm_kernel.cpp847 fVec sum_fvec = fVec(opmath_t(0)); in batch_norm_cpu_collect_stats_contiguous_internal() local
1041 fVec sum_fvec{0}, dotp_fvec{0}; in batch_norm_cpu_backward_contiguous_internal() local
H A DSoftMaxKernel.cpp183 fVec sum_fvec = fVec(float(0)); in _vec_softmax_lastdim() local