/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__.pyi (336B)
from .data_parallel import DataParallel as DataParallel, data_parallel as data_parallel from .distributed import DistributedDataParallel as DistributedDataParallel from .parallel_apply import parallel_apply as parallel_apply from .replicate import replicate as replicate from .scatter_gather import gather as gather, scatter as scatter