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 ..modules import Module from typing import Any, Optional from .common_types import _devices_t, _device_t class DistributedDataParallel(Module): process_group: Any = ... dim: int = ... module: Module = ... device_ids: _devices_t = ... output_device: _device_t = ... broadcast_buffers: bool = ... check_reduction: bool = ... broadcast_bucket_size: float = ... bucket_bytes_cap: float = ... # TODO type process_group once `distributed` module is stubbed def __init__(self, module: Module, device_ids: Optional[_devices_t] = ..., output_device: Optional[_device_t] = ..., dim: int = ..., broadcast_buffers: bool = ..., process_group: Optional[Any] = ..., bucket_cap_mb: float = ..., find_unused_parameters: bool = ..., check_reduction: bool = ...) -> None: ...