Searched refs:grouped_reduced (Results 1 – 1 of 1) sorted by relevance
430 def _ungroup_and_make_mirrored(grouped_reduced, argument454 index = [[] for _ in range(len(grouped_reduced[0]))]455 for per_replica_reduced in grouped_reduced: