Searched refs:build_nccl_then_shuffle (Results 1 – 2 of 2) sorted by relevance
402 summed_grads = all_reduce.build_nccl_then_shuffle(
788 def build_nccl_then_shuffle(input_tensors, gather_devices, nccl_red_op, function