netrans/bin/torch/nn/parallel/replicate.pyi

10 lines
269 B
Python

from typing import List, Union, Sequence, TypeVar
from ..modules import Module
from .common_types import _devices_t
T = TypeVar('T')
def replicate(network: Module[T], devices: Union[_devices_t, Sequence[_devices_t]], detach: bool = ...) -> List[
Module[T]]: ...