Searched defs:input_p_reshaped (Results 1 – 1 of 1) sorted by relevance
6298 auto input_p_reshaped = input_p.view({1, N * groups, N ? -1 : 1}); in group_norm_jvp() local6355 auto input_p_reshaped = input_p.view(view_shape); in group_norm_invstd_jvp() local