Fix unit tests

apache · Oct 17, 2022 · 2ffad17 · 2ffad17
1 parent b38d527
commit 2ffad17
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/tests/python/relay/strategy/arm_cpu/test_depthwise_conv2d.py b/tests/python/relay/strategy/arm_cpu/test_depthwise_conv2d.py
@@ -57,14 +57,14 @@ class TestDepthwiseConv2d_NHWC_HWOI(DepthwiseConv2dTests):
     """This test is for depthwise_conv2d_nhwc.generic schedule."""
 
     data_shape, kernel_size, num_filter, strides, padding, dilation = parameters(
-        ((1, 32, 32, 16), 16, (3, 3), 16, 1, 0, 1),
-        ((1, 32, 10, 16), 16, (3, 3), 16, 1, 0, 1),
-        ((1, 49, 10, 64), 64, (10, 4), 64, (2, 1), (4, 1, 5, 1), 1),
-        ((1, 32, 32, 16), 16, (3, 3), 16, 1, (0, 2, 2, 0), 1),
-        ((1, 32, 32, 16), 16, (3, 3), 16, 1, 0, 1),
-        ((1, 32, 32, 16), 16, (3, 3), 16, 1, 0, 1),
-        ((1, 32, 32, 16), 16, (3, 3), 16, 1, (0, 2, 2, 0), 2),
-        ((1, 32, 32, 16), 16, (3, 3), 16, 1, (1, 1, 2, 2), 2),
+        ((1, 32, 32, 16), (3, 3), 16, 1, 0, 1),
+        ((1, 32, 10, 16), (3, 3), 16, 1, 0, 1),
+        ((1, 49, 10, 64), (10, 4), 64, (2, 1), (4, 1, 5, 1), 1),
+        ((1, 32, 32, 16), (3, 3), 16, 1, (0, 2, 2, 0), 1),
+        ((1, 32, 32, 16), (3, 3), 16, 1, 0, 1),
+        ((1, 32, 32, 16), (3, 3), 16, 1, 0, 1),
+        ((1, 32, 32, 16), (3, 3), 16, 1, (0, 2, 2, 0), 2),
+        ((1, 32, 32, 16), (3, 3), 16, 1, (1, 1, 2, 2), 2),
     )
 
     in_dtype = parameter("int8", "int16")