File tree Expand file tree Collapse file tree 1 file changed +14
-0
lines changed
backends/arm/operator_support Expand file tree Collapse file tree 1 file changed +14
-0
lines changed Original file line number Diff line number Diff line change 22#
33# This source code is licensed under the BSD-style license found in the
44# LICENSE file in the root directory of this source tree.
5+ """Declare operator support for dim-order clone in TOSA.
6+
7+ This module registers a support check for ``dim_order_ops._clone_dim_order``
8+ ensuring input/output dtypes match and the value types are FakeTensors.
9+
10+ """
511
612import logging
713
1925
2026@register_tosa_support_check
2127class CloneSupported (SupportedTOSAOperatorCheck ):
28+ """Provide TOSA support check for ``_clone_dim_order``."""
29+
2230 targets = [exir_ops .edge .dim_order_ops ._clone_dim_order .default ]
2331
2432 tosa_specs = [
@@ -29,6 +37,12 @@ class CloneSupported(SupportedTOSAOperatorCheck):
2937 def is_node_tosa_supported (
3038 self , node : fx .Node , tosa_spec : TosaSpecification
3139 ) -> bool :
40+ """Return True if the node is supported by TOSA.
41+
42+ Verify the operator target, the number and types of inputs/outputs, and
43+ check that input and output dtypes match.
44+
45+ """
3246 if node .target not in self .targets :
3347 self .reporter .report_reject (node , f"Target { node .target } is not supported." )
3448 return False
You can’t perform that action at this time.
0 commit comments