Searched refs:rstd_tensor (Results 1 – 1 of 1) sorted by relevance
4955 const Tensor rstd_tensor = rstd.reshape_symint({N, G, 1, 1}); in infinitely_differentiable_native_group_norm_backward() local4971 ((rstd_tensor * rstd_tensor).reciprocal_() - eps).clamp_min(0); in infinitely_differentiable_native_group_norm_backward()4972 const Tensor rstd_cube = rstd_tensor * rstd_tensor * rstd_tensor; in infinitely_differentiable_native_group_norm_backward()4979 isDefined(gamma) ? rstd_tensor * gamma_tensor : rstd_tensor; in infinitely_differentiable_native_group_norm_backward()4985 c = -b * mean_tensor - c * rstd_tensor * std::move(s); in infinitely_differentiable_native_group_norm_backward()5009 dgamma = ((ds - db * mean_tensor) * rstd_tensor) in infinitely_differentiable_native_group_norm_backward()