fix shape

kendryte · Nov 28, 2023 · 7fa6bac · 7fa6bac
1 parent 748ade1
commit 7fa6bac
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/importer/onnx_/basic/test_softmax.py b/tests/importer/onnx_/basic/test_softmax.py
@@ -63,9 +63,8 @@ def _make_module(in_shape, axis, op_version):
 
 
 in_shapes = [
-    [2, 32, 3, 3],
-    [2, 32, 64, 128],
-    [1, 113, 228, 65],
+    [2, 32, 16, 4],
+    [5, 13, 28, 64],
 ]
 
 axes = [