mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
c8166d4b58
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/45350 Reviewed By: walterddr Differential Revision: D23935750 Pulled By: malfet fbshipit-source-id: 5a7d2d4fbc976699d80bb5caf4727c19fa2c5bc8
6 lines
293 B
Python
6 lines
293 B
Python
# 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']
|