Searched defs:replica_grads (Results 1 – 1 of 1) sorted by relevance
41 def aggregate_gradients_using_nccl(replica_grads): argument55 def aggregate_gradients_using_hierarchical_copy(avail_devices, replica_grads): argument