Searched refs:replica_grads (Results 1 – 1 of 1) sorted by relevance
41 def aggregate_gradients_using_nccl(replica_grads): argument44 for single_g_and_v in zip(*replica_grads):55 def aggregate_gradients_using_hierarchical_copy(avail_devices, replica_grads): argument84 for i, single_grads in enumerate(zip(*replica_grads)):