Searched refs:build_shuffle_then_shuffle (Results 1 – 2 of 2) sorted by relevance
406 summed_grads = all_reduce.build_shuffle_then_shuffle(
851 def build_shuffle_then_shuffle(input_tensors, first_gather_devices, function