parallel_apply.pyi 293 B

1234567
  1. from typing import Any, Optional, Sequence, List
  2. from .common_types import _devices_t
  3. from ..modules import Module
  4. def parallel_apply(modules: Sequence[Module], inputs: Sequence[Any], kwargs_tup: Optional[Any] = ...,
  5. devices: Optional[_devices_t] = ...) -> List[Any]: ...