Searched defs:input_t_reshaped (Results 1 – 1 of 1) sorted by relevance
6297 auto input_t_reshaped = input_t.view({1, N * groups, N ? -1 : 1}); in group_norm_jvp() local6340 auto input_t_reshaped = input_t.view(view_shape); in group_norm_mean_jvp() local6354 auto input_t_reshaped = input_t.view(view_shape); in group_norm_invstd_jvp() local