Skip to content

Commit 3db968c

Browse files
committed
Arm(R) Ethos(TM)-U NPU Relay passes and Conv2D op
*missed comment split_o Change-Id: I4a4b19ff2cd18e8f568a63ae827f44358ed85b8e
1 parent 0110b44 commit 3db968c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/python/contrib/test_ethosu/test_legalize.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -96,10 +96,10 @@ def test_split_sections_legalize():
9696
def create_graph(axis, sections):
9797
x = relay.var("x", shape=(1, 50, 50, 3))
9898
x_abs = relay.abs(x)
99-
split_o = relay.split(x_abs, sections, axis).tuple_value
99+
split_output = relay.split(x_abs, sections, axis).tuple_value
100100
outputs = list()
101101
for section_idx in range(sections):
102-
split_single_out = relay.TupleGetItem(split_o, section_idx)
102+
split_single_out = relay.TupleGetItem(split_output, section_idx)
103103
tanh = relay.tanh(split_single_out)
104104
outputs.append(tanh)
105105
tuple_out = relay.Tuple(outputs)

0 commit comments

Comments
 (0)