Searched refs:aggregate_gradients_using_hierarchical_copy (Results 1 – 2 of 2) sorted by relevance
55 def aggregate_gradients_using_hierarchical_copy(avail_devices, replica_grads): function
900 cross_device_utils.aggregate_gradients_using_hierarchical_copy(