Searched refs:norm_two_reduce_step (Results 1 – 1 of 1) sorted by relevance
179 inline void norm_two_reduce_step(Vectorized<acc_t>& acc_vec, Vectorized<scalar_t>& data_vec) { in norm_two_reduce_step() function184 inline void norm_two_reduce_step(Vectorized<float>& acc_fvec, Vectorized<BFloat16>& data_bvec) { in norm_two_reduce_step() function247 norm_two_reduce_step(acc_vec, data_vec); in norm_kernel_tensor_iterator_impl()