mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-08 07:39:33 +01:00
bfeff1eb8f
Summary: Closes https://github.com/pytorch/pytorch/issues/18724 Pull Request resolved: https://github.com/pytorch/pytorch/pull/19089 Differential Revision: D16073654 Pulled By: ezyang fbshipit-source-id: 5642179651ce45ab7c5a46cc1fcc4fd6b37fa71c
10 lines
269 B
Python
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]]: ...
|