| /aosp_15_r20/external/pytorch/torchgen/static_runtime/ |
| H A D | generator.py | 17 NativeFunctionsViewGroup,
242 def is_supported(g: NativeFunctionsGroup | NativeFunctionsViewGroup) -> bool: argument
245 if isinstance(g, NativeFunctionsViewGroup):
264 if isinstance(g, NativeFunctionsViewGroup):
537 g: NativeFunctionsViewGroup, backend_index: BackendIndex argument
631 self, groups: Sequence[NativeFunctionsViewGroup], backend_index: BackendIndex argument
639 assert isinstance(g, NativeFunctionsViewGroup)
682 self, g: NativeFunctionsViewGroup, backend_index: BackendIndex argument
710 def view(self, groups: Sequence[NativeFunctionsViewGroup]) -> str: argument
717 assert isinstance(g, NativeFunctionsViewGroup)
[all …]
|
| H A D | gen_static_runtime_ops.py | 12 from torchgen.model import DispatchKey, NativeFunctionsGroup, NativeFunctionsViewGroup
22 bound=Union[NativeFunctionsGroup, NativeFunctionsViewGroup],
33 def is_supported(g: NativeFunctionsGroup | NativeFunctionsViewGroup) -> bool: argument
185 if isinstance(g, NativeFunctionsViewGroup)
|
| H A D | config.py | 3 from torchgen.model import NativeFunctionsGroup, NativeFunctionsViewGroup
6 def func_name_base_str(g: NativeFunctionsGroup | NativeFunctionsViewGroup) -> str: argument
58 def is_hand_written(g: NativeFunctionsGroup | NativeFunctionsViewGroup) -> bool: argument
|
| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | context.py | 13 NativeFunctionsViewGroup,
24 NativeFunctionsViewGroup,
26 Union[NativeFunction, NativeFunctionsViewGroup],
43 g: NativeFunctionsGroup | NativeFunctionsViewGroup | NativeFunction,
51 elif isinstance(g, NativeFunctionsViewGroup):
|
| H A D | gen_functionalization_type.py | 40 NativeFunctionsViewGroup,
96 def __call__(self, g: NativeFunctionsViewGroup) -> str | None: argument
348 g: NativeFunctionsViewGroup, *, view_inplace: bool argument
760 selector: SelectiveBuilder, g: NativeFunctionsViewGroup argument
765 def emit_decl_helper(g: NativeFunctionsViewGroup) -> str | None: argument
776 g: NativeFunction | NativeFunctionsGroup | NativeFunctionsViewGroup, argument
789 if isinstance(g, NativeFunctionsViewGroup):
842 g: NativeFunction | NativeFunctionsGroup | NativeFunctionsViewGroup, argument
848 if isinstance(g, NativeFunctionsViewGroup):
|
| H A D | gen.py | 66 NativeFunctionsViewGroup,
1363 ) -> Sequence[NativeFunction | NativeFunctionsViewGroup]:
1366 ) -> list[NativeFunction | NativeFunctionsViewGroup]:
1367 funcs: list[NativeFunction | NativeFunctionsViewGroup] = []
1374 NativeFunctionsViewGroup(
2176 view_groups: Sequence[NativeFunctionsViewGroup], argument
2488 fn: NativeFunction | NativeFunctionsGroup | NativeFunctionsViewGroup, argument
2530 g: NativeFunction | NativeFunctionsGroup | NativeFunctionsViewGroup, argument
2533 g: NativeFunction | NativeFunctionsGroup | NativeFunctionsViewGroup, argument
2535 if isinstance(g, NativeFunctionsViewGroup):
[all …]
|
| H A D | model.py | 2665 class NativeFunctionsViewGroup: class
|
| /aosp_15_r20/external/pytorch/torchgen/api/types/ |
| H A D | signatures.py | 15 NativeFunctionsViewGroup,
292 g: NativeFunctionsViewGroup
310 g: NativeFunctionsViewGroup
373 g: NativeFunctionsViewGroup, *, is_reverse: bool argument
|
| /aosp_15_r20/external/pytorch/torchgen/api/ |
| H A D | functionalization.py | 21 NativeFunctionsViewGroup,
92 g: NativeFunctionsViewGroup, argument
|
| H A D | autograd.py | 16 NativeFunctionsViewGroup,
174 self, g: NativeFunctionsViewGroup argument
|
| /aosp_15_r20/external/pytorch/tools/autograd/ |
| H A D | load_derivatives.py | 46 NativeFunctionsViewGroup,
69 view_groups: list[NativeFunctionsViewGroup], argument
72 view_name_to_group: dict[OperatorName, NativeFunctionsViewGroup] = {
124 if isinstance(g, NativeFunctionsViewGroup)
|