Name	Size	Mode	Actions
__pycache__/	-	0755	rm
comm.py	10788	0644	editdlrm
common_types.pyi	123	0644	editdlrm
data_parallel.py	10270	0644	editdlrm
data_parallel.pyi	717	0644	editdlrm
distributed.py	69172	0644	editdlrm
distributed.pyi	860	0644	editdlrm
parallel_apply.py	3216	0644	editdlrm
parallel_apply.pyi	293	0644	editdlrm
replicate.py	6007	0644	editdlrm
replicate.pyi	245	0644	editdlrm
scatter_gather.py	3136	0644	editdlrm
scatter_gather.pyi	989	0644	editdlrm
_functions.py	4664	0644	editdlrm
__init__.py	645	0644	editdlrm
__init__.pyi	336	0644	editdlrm

from typing import Any, Optional from .common_types import _devices_t, _device_t from ..modules import Module from ... import device, Tensor class DataParallel(Module): module: Module = ... device_ids: _devices_t = ... dim: int = ... output_device: _device_t = ... src_device_obj: device = ... def __init__(self, module: Module, device_ids: Optional[_devices_t] = ..., output_device: Optional[_device_t] = ..., dim: int = ...) -> None: ... def data_parallel(module: Module, inputs: Any, device_ids: Optional[_devices_t] = ..., output_device: Optional[_device_t] = ..., dim: int = ..., module_kwargs: Optional[Any] = ...) -> Tensor: ...