Searched defs:rstd_val (Results 1 – 4 of 4) sorted by relevance
82 CTYPE rstd_val; in layer_norm() local
114 T rstd_val = vec::vec_reduce_all([](Vec& x, Vec& y) { return x + y; }, acc1_vec); in ColumnwiseMoments() local158 opmath_t rstd_val = vec::vec_reduce_all([](fVec& x, fVec& y) { return x + y; }, acc1_fvec); in ColumnwiseMoments() local421 opmath_t mean_val{0}, rstd_val{0}; in GroupNormKernelImplChannelsLastInternal() local452 opmath_t rstd_val = tmp_buffer_data[n * 2 * G + 2 * g + 1]; in GroupNormKernelImplChannelsLastInternal() local
246 T_ACC rstd_val = c10::cuda::compat::rsqrt(wd.sigma2 + eps); in vectorized_layer_norm_kernel_impl() local325 const T_ACC rstd_val = rstd[i1]; in compute_gI() local413 const T_ACC rstd_val = rstd[bIdx]; in layer_norm_grad_input_kernel_vectorized() local
3223 float mean_val{0}, rstd_val{0}; local3260 float rstd_val = rstd_data[n * G + g]; local