Searched refs:all_reduce_indexed_slices (Results 1 – 2 of 2) sorted by relevance
497 def all_reduce_indexed_slices( member in CollectiveReplicaLauncher
1173 launcher.all_reduce_indexed_slices(indexed_slice, options))