parallel_apply.pyi 366 Bytes
Newer Older
Mandeep Singh Baines's avatar
Mandeep Singh Baines committed
1
2
3
4
5
6
7
8
9
# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.

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]: ...