Merge branch 'main' into main

Author: matt-cossins

backends/arm/_passes/decompose_sum_pass.py

@@ -78,7 +78,7 @@ def call_operator(self, op, args, kwargs, meta):
         for dim in dims:
             input_node = super().call_operator(
                 sum_op,
-                (input_node, dim, True),
+                (input_node, [dim], True),
                 kwargs,
                 meta,
                 updated=True,

backends/arm/operators/op_sum.py

@@ -43,7 +43,7 @@ def define_node(
 
         tensor = inputs[0]
         input_shape = list(tensor.shape)
-        dim = int(inputs[1].number % len(input_shape))
+        dim = int(inputs[1].special[0] % len(input_shape))
 
         attr = ts.TosaSerializerAttribute()
         attr.ReduceSumAttribute(dim)