def METHOD_NAME(self, tensor: Tensor, group: Optional[Any] = None, sync_grads: bool = False) -> Tensor:
    """Perform a all_gather on all processes."""
    return _all_gather_ddp_if_available(tensor, group=group, sync_grads=sync_grads)