Remove is_quant_node from NodeVisitor.define_node

Signed-off-by: Oscar Andersson <[email protected]>
Change-Id: Ibb17add461dc79e022a7f4accde29f9f9d61b16d

Author: oscarandersson8218

backends/arm/operators/node_visitor.py

@@ -39,7 +39,6 @@ def define_node(
         tosa_graph: ts.TosaSerializer,
         inputs: List[TosaArg],
         output: TosaArg,
-        is_quant_node: bool,
     ) -> None:
         raise NotImplementedError("NodeVisitor must be extended.")
 

backends/arm/operators/op_add.py

@@ -38,7 +38,6 @@ def define_node(
         tosa_graph: ts.TosaSerializer,
         inputs: List[TosaArg],
         output: TosaArg,
-        is_quant_node: bool,
     ) -> None:
         # Specification (0.80.0) states that input and output types
         # should all be the same
@@ -96,15 +95,14 @@ def define_node(
         tosa_graph: ts.TosaSerializer,
         inputs: List[TosaArg],
         output: TosaArg,
-        is_quant_node: bool,
     ) -> None:
         # Specification (0.80.0) states that input and output types
         # should all be the same
         assert inputs[0].dtype == inputs[1].dtype == output.dtype
 
         if inputs[0].dtype in [ts.DType.INT8, ts.DType.INT32]:
             # Call the inherited define_node for handling integers
-            super().define_node(node, tosa_graph, inputs, output, is_quant_node)
+            super().define_node(node, tosa_graph, inputs, output)
         else:
             # FP32 Add lowering
             assert inputs[0].dtype == ts.DType.FP32

backends/arm/operators/op_avg_pool2d.py

@@ -73,7 +73,6 @@ def define_node(
         tosa_graph: ts.TosaSerializer,
         inputs: List[TosaArg],
         output: TosaArg,
-        is_quant_node: bool,
     ) -> None:
         input_tensor = inputs[0]
         assert input_tensor.dtype == ts.DType.INT8
@@ -105,14 +104,13 @@ def define_node(
         tosa_graph: ts.TosaSerializer,
         inputs: List[TosaArg],
         output: TosaArg,
-        is_quant_node: bool,
     ) -> None:
         assert (
             inputs[0].dtype == ts.DType.INT8 or inputs[0].dtype == ts.DType.FP32
         ), "Only FP32 and INT8 supported"
 
         if inputs[0].dtype == ts.DType.INT8:
-            super().define_node(node, tosa_graph, inputs, output, is_quant_node)
+            super().define_node(node, tosa_graph, inputs, output)
 
         if inputs[0].dtype == ts.DType.FP32:
             accumulator_type = ts.DType.FP32

backends/arm/operators/op_batch_norm.py

@@ -42,7 +42,6 @@ def define_node(
         tosa_graph: ts.TosaSerializer,
         inputs: List[TosaArg],
         output: TosaArg,
-        is_quant_node: bool,
     ) -> None:
         # Decompose batch norm into sequence
         (activations, weights, bias, running_mean, running_var, momentum, epsilon) = (

backends/arm/operators/op_bmm.py

@@ -35,7 +35,6 @@ def define_node(
         tosa_graph: ts.TosaSerializer,
         inputs: List[TosaArg],
         output: TosaArg,
-        is_quant_node: bool,
     ) -> None:
 
         assert inputs[0].dtype == inputs[1].dtype, "Both inputs must be of same type"

backends/arm/operators/op_cat.py

@@ -30,7 +30,6 @@ def define_node(
         tosa_graph: ts.TosaSerializer,
         inputs: List[TosaArg],
         output: TosaArg,
-        is_quant_node: bool,
     ) -> None:
 
         tensors = inputs[0].special

backends/arm/operators/op_conv2d.py

@@ -53,7 +53,6 @@ def define_node(
         tosa_graph: ts.TosaSerializer,
         inputs: List[TosaArg],
         output: TosaArg,
-        is_quant_node: bool,
     ) -> None:
         input, weight, bias, stride, pad, dilation, _, _, group = inputs
 

backends/arm/operators/op_dequant.py

@@ -1,4 +1,4 @@
-# Copyright 2023 Arm Limited and/or its affiliates.
+# Copyright 2023-2024 Arm Limited and/or its affiliates.
 #
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
@@ -29,7 +29,6 @@ def define_node(
         tosa_graph: ts.TosaSerializer,
         inputs: List[TosaArg],
         output: TosaArg,
-        is_quant_node: bool,
     ) -> None:
         item_name = inputs[0].name
         ## Simply add an identityOp

backends/arm/operators/op_exp.py

@@ -34,7 +34,6 @@ def define_node(
         tosa_graph: ts.TosaSerializer,
         inputs: List[TosaArg],
         output: TosaArg,
-        is_quant_node: bool,
     ) -> None:
 
         assert len(node.all_input_nodes) == 1

backends/arm/operators/op_full.py

@@ -31,7 +31,6 @@ def define_node(
         tosa_graph: ts.TosaSerializer,
         inputs: List[TosaArg],
         output: TosaArg,
-        is_quant_node: bool,
     ) -> None:
 
         shape = tosa_shape(inputs[0].special, output.dim_order)