minor fix

apache · Sep 13, 2024 · f697f43 · f697f43
1 parent 468c3d3
commit f697f43
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/contrib/test_msc/test_tools.py b/tests/python/contrib/test_msc/test_tools.py
@@ -229,7 +229,7 @@ def get_model_info(compile_type):
             "inputs": [
                 {"name": "input_0", "shape": [1, 3, 224, 224], "dtype": "float32", "layout": "NCHW"}
             ],
-            "outputs": [{"name": "output", "shape": [1, 1000], "dtype": "float32", "layout": "NC"}],
+            "outputs": [{"name": "output", "shape": [1, 1000], "dtype": "float32", "layout": "NW"}],
             "nodes": {
                 "total": 229,
                 "input": 1,