Home
last modified time | relevance | path

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

/external/executorch/exir/backend/test/
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
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
Dtest_backends_nested.py161 def is_node_supported(self, submodules, node: torch.fx.Node) -> bool: member in CondOperatorSupport
/external/executorch/exir/backend/test/demos/rpc/
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
/external/executorch/exir/backend/canonical_partitioners/
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
/external/executorch/backends/arm/operator_support/
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
Dmean_dim_support.py29 def is_node_supported(self, node: fx.Node, tosa_spec: TosaSpecification) -> bool: member in MeanDimSupported
Dvar_correction_support.py29 def is_node_supported(self, node: fx.Node, tosa_spec: TosaSpecification) -> bool: member in VarCorrectionSupported
/external/pytorch/torch/fx/passes/
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
/external/pytorch/torch/fx/passes/backends/
Dcudagraphs.py13 def is_node_supported(self, submodules, node: torch.fx.Node) -> bool: member in CudaGraphsSupport
/external/executorch/backends/example/
Dexample_partitioner.py38 def is_node_supported(self, _submodules, node: torch.fx.Node) -> bool: member in ExamplePartitioner.__init__.DequantQuantOperatorSupport
/external/pytorch/test/fx/
Dtest_partitioner_order.py13 def is_node_supported( member in DummyDevOperatorSupport
/external/executorch/backends/apple/coreml/partition/
Dcoreml_partitioner.py38 def is_node_supported(self, submodules, node: torch.fx.Node) -> bool: member in OperatorsSupportedForCoreMLBackend
/external/executorch/backends/mediatek/
Dpartitioner.py40 def is_node_supported(self, _, node: torch.fx.Node) -> bool: member in NeuropilotOperatorsSupport
/external/executorch/backends/qualcomm/partition/
Dqnn_partitioner.py57 def is_node_supported(self, _, node: torch.fx.Node) -> bool: member in QnnOperatorSupport
/external/executorch/backends/apple/mps/partition/
Dmps_partitioner.py43 def is_node_supported(self, submodules, node: torch.fx.Node) -> bool: member in MPSOperatorSupport
/external/executorch/backends/vulkan/partitioner/
Dvulkan_partitioner.py219 def is_node_supported( member in VulkanSupportedOperators
/external/pytorch/test/export/
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
/external/executorch/backends/qualcomm/utils/
Dutils.py139 def is_node_supported(self, _, node: torch.fx.Node) -> bool: member in _AnnotationSkipper
/external/pytorch/torch/onnx/_internal/
Donnxruntime.py195 def is_node_supported( member in OrtOperatorSupport
/external/pytorch/test/
Dtest_fx_passes.py244 def is_node_supported(self, submodules, node: torch.fx.Node) -> bool: member in MockOperatorSupport