/usr/local/lib64/python3.6/site-packages/torch/nn/parallel
NameSizeModeActions
__pycache__/-0755rm
comm.py107880644editdlrm
common_types.pyi1230644editdlrm
data_parallel.py102700644editdlrm
data_parallel.pyi7170644editdlrm
distributed.py691720644editdlrm
distributed.pyi8600644editdlrm
parallel_apply.py32160644editdlrm
parallel_apply.pyi2930644editdlrm
replicate.py60070644editdlrm
replicate.pyi2450644editdlrm
scatter_gather.py31360644editdlrm
scatter_gather.pyi9890644editdlrm
_functions.py46640644editdlrm
__init__.py6450644editdlrm
__init__.pyi3360644editdlrm
Edit: /usr/local/lib64/python3.6/site-packages/torch/nn/parallel/__init__.py (645B)
from .parallel_apply import parallel_apply from .replicate import replicate from .data_parallel import DataParallel, data_parallel from .scatter_gather import scatter, gather from .distributed import DistributedDataParallel __all__ = ['replicate', 'scatter', 'parallel_apply', 'gather', 'data_parallel', 'DataParallel', 'DistributedDataParallel'] def DistributedDataParallelCPU(*args, **kwargs): import warnings warnings.warn("torch.nn.parallel.DistributedDataParallelCPU is deprecated, " "please use torch.nn.parallel.DistributedDataParallel instead.") return DistributedDataParallel(*args, **kwargs)