Searched defs:expanded_grad_weight (Results 1 – 1 of 1) sorted by relevance
263 const auto expanded_grad_weight = normalized_input * grad_out.transpose(0, 1); in batch_norm_backward_plumbing() local414 const auto expanded_grad_weight = reshape_dim_into(1, 1, normalized_input) * grad_out; in native_group_norm_backward_plumbing() local625 const auto expanded_grad_weight = normalized_input * grad_out; in native_layer_norm_backward_plumbing() local