Searched refs:aggregate_gradients_using_nccl (Results 1 – 2 of 2) sorted by relevance
41 def aggregate_gradients_using_nccl(replica_grads): function
894 reduced = cross_device_utils.aggregate_gradients_using_nccl(