Searched defs:data_bvec (Results 1 – 5 of 5) sorted by relevance
755 bVec data_bvec = bVec::loadu(input_ptr + d); in batch_norm_cpu_contiguous_impl() local814 bVec data_bvec = bVec::loadu(input_ptr + d); in batch_norm_cpu_channels_last_impl() local852 bVec data_bvec = bVec::loadu(input_ptr + d); in batch_norm_cpu_collect_stats_contiguous_internal() local873 bVec data_bvec = bVec::loadu(input_ptr + d); in batch_norm_cpu_collect_stats_contiguous_internal() local927 bVec data_bvec = bVec::loadu(input_ptr + d); in batch_norm_cpu_collect_stats_channels_last_internal() local957 bVec data_bvec = bVec::loadu(input_ptr + d); in batch_norm_cpu_collect_stats_channels_last_internal() local
196 Vec data_bvec = Vec::loadu(X_ptr + d); in CalcMeanVar() local212 Vec data_bvec = Vec::loadu(X_ptr + d, C - d); in CalcMeanVar() local260 Vec data_bvec = Vec::loadu(X_ptr + d); in ApplyScaleBias() local271 Vec data_bvec = Vec::loadu(X_ptr + d, C - d); in ApplyScaleBias() local
220 bVec data_bvec = bVec::loadu(in + d2); in cpu_adaptive_avg_pool2d_channels_last() local644 bVec data_bvec = bVec::loadu(in + d2); in cpu_adaptive_avg_pool3d_channels_last() local
311 bVec data_bvec = bVec::loadu(in + d2); in cpu_avg_pool2d_channels_last() local870 bVec data_bvec = bVec::loadu(in + d2); in cpu_avg_pool3d_channels_last() local
184 inline void norm_two_reduce_step(Vectorized<float>& acc_fvec, Vectorized<BFloat16>& data_bvec) { in norm_two_reduce_step()