Arm backend: Refactor is_consumer_node_dw_conv2d

Refactor is_consumer_node_depthwise_conv2d to not use TosaArg.

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

Author: oscarandersson8218

backends/arm/tosa_utils.py

@@ -153,14 +153,14 @@ def get_new_shape(l_rank_in, h_rank_in):
         return reshaped, input2
 
 
-def is_consumer_node_depthwise_conv2d(node):
+def is_consumer_node_depthwise_conv2d(node: Node):
     consumer_node = list(node.users)[0]
     if consumer_node.target == exir_ops.edge.aten.convolution.default:
-        inputs = getNodeArgs(consumer_node)
-        group = inputs[-1]
-        in_channels = inputs[0].shape[1]
-        out_channels = inputs[1].shape[0]
-        if (in_channels == group.number) and (out_channels % in_channels) == 0:
+        consumer_node_inputs = consumer_node.all_input_nodes
+        groups = consumer_node.args[-1]
+        in_channels = consumer_node_inputs[0].meta["val"].shape[1]
+        out_channels = consumer_node_inputs[1].meta["val"].shape[0]
+        if (in_channels == groups) and (out_channels % in_channels) == 0:
             return True
 
     return False