Searched defs:_all_gather (Results 1 – 2 of 2) sorted by relevance
196 def _all_gather(obj, worker_names=None, timeout: float = UNSET_RPC_TIMEOUT): function
363 def _all_gather(self, input_tensor: core.TensorLike, member in CollectiveReplicaLauncher