Spaces:
Running
Running
File size: 361 Bytes
c61ccee |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
# The functions here have been moved to torch.nn.parallel.comm
from torch.nn.parallel.comm import (
broadcast,
broadcast_coalesced,
gather,
reduce_add,
reduce_add_coalesced,
scatter,
)
__all__ = [
"broadcast",
"broadcast_coalesced",
"reduce_add",
"reduce_add_coalesced",
"scatter",
"gather",
]
|