| /external/pytorch/torch/fx/passes/ |
| D | operator_support.py | 29 class OperatorSupportBase(abc.ABC): class
39 class OperatorSupport(OperatorSupportBase):
136 def create_op_support(is_node_supported: IsNodeSupported) -> OperatorSupportBase:
142 class FunctionalOperatorSupport(OperatorSupportBase):
151 def chain(*op_support: OperatorSupportBase) -> OperatorSupportBase: argument
165 def any_chain(*op_support: OperatorSupportBase) -> OperatorSupportBase: argument
184 def decline_if_input_dtype(cls, dtype: torch.dtype) -> OperatorSupportBase:
199 def decline_if_node_in_names(cls, disallow_set: t.Set[str]) -> OperatorSupportBase:
|
| D | splitter_base.py | 16 OperatorSupportBase,
109 operator_support: OperatorSupportBase, argument
317 operator_support: OperatorSupportBase, argument
|
| /external/executorch/exir/backend/canonical_partitioners/ |
| D | pattern_op_partitioner.py | 12 from torch.fx.passes.operator_support import any_chain, OperatorSupportBase
19 op_support: Optional[OperatorSupportBase] = None, argument
21 final_op_support: Optional[OperatorSupportBase] = op_support
29 class MatchTag(OperatorSupportBase):
63 op_support: Optional[OperatorSupportBase] = None, argument
86 final_op_support: Optional[OperatorSupportBase] = op_support
110 class MatchTag(OperatorSupportBase):
|
| /external/executorch/exir/backend/test/demos/rpc/ |
| D | executor_backend_partitioner.py | 26 from torch.fx.passes.operator_support import any_chain, OperatorSupportBase
29 class AnyOperatorSupport(OperatorSupportBase):
34 class AnyDelegateSupport(OperatorSupportBase):
|
| /external/executorch/exir/backend/test/ |
| D | op_partitioner_demo.py | 32 from torch.fx.passes.operator_support import any_chain, OperatorSupportBase
35 class AllOperatorSupport(OperatorSupportBase):
40 class AddOperatorSupport(OperatorSupportBase):
47 class MatmulOperatorSupport(OperatorSupportBase):
192 class OpsToNotDecomposeOperatorSupport(OperatorSupportBase):
|
| D | test_graph_partition.py | 19 from torch.fx.passes.operator_support import OperatorSupportBase
71 op_support: Optional[OperatorSupportBase] = None, argument
182 class TestOperatorSupport(OperatorSupportBase):
|
| D | test_backends_nested.py | 37 from torch.fx.passes.operator_support import any_chain, OperatorSupportBase
160 class CondOperatorSupport(OperatorSupportBase):
|
| /external/executorch/backends/apple/coreml/partition/ |
| D | coreml_partitioner.py | 23 from torch.fx.passes.operator_support import OperatorSupportBase
29 class OperatorsSupportedForCoreMLBackend(OperatorSupportBase):
|
| /external/executorch/backends/example/ |
| D | example_partitioner.py | 24 from torch.fx.passes.operator_support import OperatorSupportBase
37 class DequantQuantOperatorSupport(OperatorSupportBase):
|
| /external/executorch/backends/mediatek/ |
| D | partitioner.py | 22 from torch.fx.passes.operator_support import OperatorSupportBase
25 class NeuropilotOperatorsSupport(OperatorSupportBase):
|
| /external/executorch/backends/apple/mps/partition/ |
| D | mps_partitioner.py | 26 from torch.fx.passes.operator_support import OperatorSupportBase
38 class MPSOperatorSupport(OperatorSupportBase):
|
| /external/executorch/backends/qualcomm/partition/ |
| D | qnn_partitioner.py | 28 from torch.fx.passes.operator_support import OperatorSupportBase
38 class QnnOperatorSupport(OperatorSupportBase):
|
| /external/executorch/backends/arm/operator_support/ |
| D | tosa_supported_operators.py | 14 from torch.fx.passes.operator_support import OperatorSupportBase
69 class TOSASupportedOperators(OperatorSupportBase):
|
| /external/executorch/docs/source/ |
| D | compiler-custom-compiler-passes.md | 390 * `operator_support (OperatorSupportBase)`: The object used to determine if a
402 [`OperatorSupportBase`](https://github.com/pytorch/pytorch/blob/8597d37536ef11bdf6b0a539ab79af876e1…
408 returns False if any of the OperatorSupportBase return False) and
410 (which returns True if any of the OperatorSupportBase returns True).
416 from torch.fx.passes.operator_support import any_chain, OperatorSupportBase
418 class AddMulOperatorSupport(OperatorSupportBase):
452 * `op_support (OperatorSupportBase)`: A OperatorSupportBase that can be created
457 * Multiple OperatorSupportBase classes chained together with `chain()` or `any_chain()`
461 supported by the union of the given OperatorSupportBase object and the
|
| /external/pytorch/docs/source/ |
| D | torch.compiler_transformations.rst | 393 - ``operator_support (OperatorSupportBase)``: The object used to
405 ```OperatorSupportBase`` <https://github.com/pytorch/pytorch/blob/main/torch/fx/passes/operator_sup…
409 ``OperatorSupportBase`` by using
411 returns False if any of the OperatorSupportBase return False) and
413 (which returns True if any of the OperatorSupportBase returns True).
420 from torch.fx.passes.operator_support import any_chain, OperatorSupportBase
422 class AddMulOperatorSupport(OperatorSupportBase):
|
| /external/executorch/backends/vulkan/partitioner/ |
| D | vulkan_partitioner.py | 41 from torch.fx.passes.operator_support import OperatorSupportBase
52 class VulkanSupportedOperators(OperatorSupportBase):
|
| /external/pytorch/torch/fx/passes/infra/ |
| D | partitioner.py | 12 from torch.fx.passes.operator_support import OperatorSupportBase
62 operator_support: OperatorSupportBase, argument
|
| /external/executorch/backends/qualcomm/utils/ |
| D | utils.py | 94 from torch.fx.passes.operator_support import OperatorSupportBase
98 class _AnnotationSkipper(OperatorSupportBase):
|