Arm(R) Ethos(TM)-U NPU Relay passes and Conv2D op

*missed comment split_o Change-Id: I4a4b19ff2cd18e8f568a63ae827f44358ed85b8e
apache · Sep 3, 2021 · f780b9e · f780b9e
1 parent 0110b44
commit f780b9e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/python/contrib/test_ethosu/test_legalize.py b/tests/python/contrib/test_ethosu/test_legalize.py
@@ -96,10 +96,10 @@ def test_split_sections_legalize():
     def create_graph(axis, sections):
         x = relay.var("x", shape=(1, 50, 50, 3))
         x_abs = relay.abs(x)
-        split_o = relay.split(x_abs, sections, axis).tuple_value
+        split_output = relay.split(x_abs, sections, axis).tuple_value
         outputs = list()
         for section_idx in range(sections):
-            split_single_out = relay.TupleGetItem(split_o, section_idx)
+            split_single_out = relay.TupleGetItem(split_output, section_idx)
             tanh = relay.tanh(split_single_out)
             outputs.append(tanh)
         tuple_out = relay.Tuple(outputs)