Home
last modified time | relevance | path

Searched refs:rstd_tensor (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A DFunctionsManual.cpp4955 const Tensor rstd_tensor = rstd.reshape_symint({N, G, 1, 1}); in infinitely_differentiable_native_group_norm_backward() local
4971 ((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()