|
8 | 8 | import warnings |
9 | 9 |
|
10 | 10 | class MultiStreamModuleHint(object): |
| 11 | + r""" |
| 12 | + MultiStreamModuleHint is a hint to MultiStreamModule about how to split the inputs |
| 13 | + or concat the output. Each argument should be None, with type of int or a container |
| 14 | + which containes int or None such as: (0, None, ...) or [0, None, ...]. If the argument |
| 15 | + is None, it means this argument will not be split or concat. If the argument is with |
| 16 | + type int, its value means along which dim this argument will be split or concat. |
| 17 | +
|
| 18 | + Args: |
| 19 | + *args: Variable length argument list. |
| 20 | + **kwargs: Arbitrary keyword arguments. |
| 21 | +
|
| 22 | + Returns: |
| 23 | + intel_extension_for_pytorch.cpu.runtime.MultiStreamModuleHint: Generated |
| 24 | + intel_extension_for_pytorch.cpu.runtime.MultiStreamModuleHint object. |
| 25 | +
|
| 26 | + :meta public: |
| 27 | + """ |
| 28 | + |
11 | 29 | def __init__(self, *args, **kwargs): |
12 | | - r""" |
13 | | - MultiStreamModuleHint is a hint to MultiStreamModule about how to split the inputs |
14 | | - or concat the output. Each argument should be None, with type of int or a container |
15 | | - which containes int or None such as: (0, None, ...) or [0, None, ...]. If the argument |
16 | | - is None, it means this argument will not be split or concat. If the argument is with |
17 | | - type int, its value means along which dim this argument will be split or concat. |
18 | | -
|
19 | | - Args: |
20 | | - *args: Variable length argument list. |
21 | | - **kwargs: Arbitrary keyword arguments. |
22 | | -
|
23 | | - Returns: |
24 | | - intel_extension_for_pytorch.cpu.runtime.MultiStreamModuleHint: Generated |
25 | | - intel_extension_for_pytorch.cpu.runtime.MultiStreamModuleHint object. |
26 | | -
|
27 | | - :meta public: |
28 | | - """ |
29 | 30 | self.args = list(args) |
30 | 31 | self.kwargs = kwargs |
31 | 32 | self.args_len = args.__len__() |
|
0 commit comments