Repository URL to install this package:
Version:
2.1.2+cpu ▾
|
# 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",
]