Searched refs:all_reduce_indexed_slices (Results 1 – 2 of 2) sorted by relevance
504 def all_reduce_indexed_slices(self, member in CollectiveReplicaLauncher
1288 outputs.append(self._launchers[i].all_reduce_indexed_slices(