Add as an attribute

sfvaroglu · sfvaroglu · commit 8d86cd368ed7 · 2022-04-04T11:46:07.000-07:00
diff --git a/tests/python/contrib/test_arm_compute_lib/test_add.py b/tests/python/contrib/test_arm_compute_lib/test_add.py
@@ -75,8 +75,8 @@ def _get_expected_codegen(shape, dtype, op_name, qnn_params):
     }
 
     if qnn_params:
-        node["lhs_axis"] = [[-1]]
-        node["rhs_axis"] = [[-1]]
+        node["attrs"]["lhs_axis"] = [[-1]]
+        node["attrs"]["rhs_axis"] = [[-1]]
 
     return [*inputs, node]
 

Original file line number	Diff line number	Diff line change
`@@ -75,8 +75,8 @@ def _get_expected_codegen(shape, dtype, op_name, qnn_params):`
`75`	`75`	`}`
`76`	`76`
`77`	`77`	`if qnn_params:`
`78`		`- node["lhs_axis"] = [[-1]]`
`79`		`- node["rhs_axis"] = [[-1]]`
	`78`	`+ node["attrs"]["lhs_axis"] = [[-1]]`
	`79`	`+ node["attrs"]["rhs_axis"] = [[-1]]`
`80`	`80`
`81`	`81`	`return [*inputs, node]`
`82`	`82`