# The functions here have been moved to torch.nn.parallel.comm | |
from torch.nn.parallel.comm import broadcast, broadcast_coalesced, reduce_add, \ | |
reduce_add_coalesced, scatter, gather | |
__all__ = [broadcast, broadcast_coalesced, reduce_add, reduce_add_coalesced, scatter, gather] |