torch/nn/parallel/replicate.pyi

from typing import List, Union, Sequence, TypeVar
from ..modules import Module
from .common_types import _devices_t


def replicate(network: Module, devices: Union[_devices_t, Sequence[_devices_t]], detach: bool = ...) -> List[
    Module]: ...