Home
last modified time | relevance | path

Searched defs:is_node_supported (Results 1 – 21 of 21) sorted by relevance

/aosp_15_r20/external/executorch/exir/backend/test/
H A Dop_partitioner_demo.py36 def is_node_supported(self, submodules, node: torch.fx.Node) -> bool: member in AllOperatorSupport
41 def is_node_supported(self, submodules, node: torch.fx.Node) -> bool: member in AddOperatorSupport
48 def is_node_supported(self, submodules, node: torch.fx.Node) -> bool: member in MatmulOperatorSupport
193 def is_node_supported(self, submodules, node: torch.fx.Node) -> bool: member in OpsToNotDecomposeOperatorSupport
H A Dtest_graph_partition.py183 def is_node_supported(self, submodules, node: torch.fx.Node) -> bool: member in TestGraphPartition.test_graph_partition_with_op_support.TestOperatorSupport
H A Dtest_backends_nested.py161 def is_node_supported(self, submodules, node: torch.fx.Node) -> bool: member in CondOperatorSupport
/aosp_15_r20/external/pytorch/torch/fx/passes/
H A Doperator_support.py32 def is_node_supported( member in OperatorSupportBase
66 def is_node_supported( member in OperatorSupport
143 def is_node_supported( member in create_op_support.FunctionalOperatorSupport
/aosp_15_r20/external/executorch/exir/backend/test/demos/rpc/
H A Dexecutor_backend_partitioner.py30 def is_node_supported(self, submodules, node: torch.fx.Node) -> bool: member in AnyOperatorSupport
35 def is_node_supported(self, submodules, node: torch.fx.Node) -> bool: member in AnyDelegateSupport
/aosp_15_r20/external/executorch/exir/backend/canonical_partitioners/
H A Dpattern_op_partitioner.py30 def is_node_supported(self, submodules, node: torch.fx.Node) -> bool: member in generate_partitions_from_list_of_nodes.MatchTag
111 def is_node_supported(self, submodules, node: torch.fx.Node) -> bool: member in generate_pattern_op_partitions.MatchTag
/aosp_15_r20/external/executorch/backends/arm/operator_support/
H A Dtosa_supported_operators.py26 def is_node_supported(self, node: fx.Node, tosa_spec: TosaSpecification) -> bool: member in SupportedTOSAOperatorCheck
74 def is_node_supported(self, submodules, node: fx.Node) -> bool: member in TOSASupportedOperators
H A Dmean_dim_support.py29 def is_node_supported(self, node: fx.Node, tosa_spec: TosaSpecification) -> bool: member in MeanDimSupported
H A Dvar_correction_support.py29 def is_node_supported(self, node: fx.Node, tosa_spec: TosaSpecification) -> bool: member in VarCorrectionSupported
/aosp_15_r20/external/pytorch/torch/fx/passes/backends/
H A Dcudagraphs.py13 def is_node_supported(self, submodules, node: torch.fx.Node) -> bool: member in CudaGraphsSupport
/aosp_15_r20/external/executorch/backends/example/
H A Dexample_partitioner.py38 def is_node_supported(self, _submodules, node: torch.fx.Node) -> bool: member in ExamplePartitioner.__init__.DequantQuantOperatorSupport
/aosp_15_r20/external/pytorch/test/fx/
H A Dtest_partitioner_order.py13 def is_node_supported( member in DummyDevOperatorSupport
/aosp_15_r20/external/executorch/backends/apple/coreml/partition/
H A Dcoreml_partitioner.py38 def is_node_supported(self, submodules, node: torch.fx.Node) -> bool: member in OperatorsSupportedForCoreMLBackend
/aosp_15_r20/external/executorch/backends/mediatek/
H A Dpartitioner.py40 def is_node_supported(self, _, node: torch.fx.Node) -> bool: member in NeuropilotOperatorsSupport
/aosp_15_r20/external/executorch/backends/qualcomm/partition/
H A Dqnn_partitioner.py57 def is_node_supported(self, _, node: torch.fx.Node) -> bool: member in QnnOperatorSupport
/aosp_15_r20/external/pytorch/test/export/
H A Dtest_passes.py69 def is_node_supported(self, submodules, node: torch.fx.Node) -> bool: member in _AddOperatorSupport
74 def is_node_supported(self, submodules, node: torch.fx.Node) -> bool: member in _AtenAddOperatorSupport
/aosp_15_r20/external/executorch/backends/apple/mps/partition/
H A Dmps_partitioner.py43 def is_node_supported(self, submodules, node: torch.fx.Node) -> bool: member in MPSOperatorSupport
/aosp_15_r20/external/executorch/backends/vulkan/partitioner/
H A Dvulkan_partitioner.py219 def is_node_supported( member in VulkanSupportedOperators
/aosp_15_r20/external/executorch/backends/qualcomm/utils/
H A Dutils.py139 def is_node_supported(self, _, node: torch.fx.Node) -> bool: member in _AnnotationSkipper
/aosp_15_r20/external/pytorch/torch/onnx/_internal/
H A Donnxruntime.py195 def is_node_supported( member in OrtOperatorSupport
/aosp_15_r20/external/pytorch/test/
H A Dtest_fx_passes.py244 def is_node_supported(self, submodules, node: torch.fx.Node) -> bool: member in MockOperatorSupport