Searched defs:num_front_dims_to_reduce (Results 1 – 1 of 1) sorted by relevance
615 const auto num_front_dims_to_reduce = grad_out.dim() - normalized_shape.size(); in native_layer_norm_backward_plumbing() local626 const auto num_front_dims_to_reduce = in native_layer_norm_backward_plumbing() local