JSX_TTS / torch /nn /parallel /parallel_apply.pyi
UMMJ's picture
Upload 5875 files
9dd3461
from typing import Any, Optional, Sequence, List
from .common_types import _devices_t
from ..modules import Module
def parallel_apply(modules: Sequence[Module], inputs: Sequence[Any], kwargs_tup: Optional[Any] = ...,
devices: Optional[_devices_t] = ...) -> List[Any]: ...