Skip to content

Commit c53ab9f

Browse files
committed
Arm backend: Fix for test
Change-Id: I5a25a990c9aa2f762a882be3f958df86e40f550e
1 parent 82358bd commit c53ab9f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

backends/arm/test/misc/test_outputs_order.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ def test_network_output_order_and_restore(tmp_path, batch_size):
112112
out_shapes = _read_tosa_outputs(tosa_files[0])
113113
# We use shape that is unique to output to check
114114
# that we preserve output order
115-
channel_dims = [s[-1] for s in out_shapes]
115+
channel_dims = [s[1] for s in reversed(out_shapes)]
116116
assert channel_dims == [1, 2, 3], (
117117
"Outputs in .tosa do not keep author order: "
118118
f"expected [1, 2, 3], got {channel_dims}"

0 commit comments

Comments
 (0)