Fix test cases

apache · Aug 31, 2022 · 2533396 · 2533396
1 parent a04e4fb
commit 2533396
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/python/relay/strategy/arm_cpu/test_depthwise_conv2d.py b/tests/python/relay/strategy/arm_cpu/test_depthwise_conv2d.py
@@ -151,8 +151,9 @@ class TestDepthwiseConv2d_NHWC_HWOI_DSP(BasicDepthwiseConv2dTests):
     """This test is for depthwise_conv2d_nhwc.generic schedule."""
 
     data_shape, kernel_size, num_filter, strides, padding, dilation = tvm.testing.parameters(
+        # Depthwise_conv2d parameters from MobileNetV1 0.25x
         ((1, 48, 48, 8),  (3, 3), 8,  (1, 1), 1, 1),
-        ((1, 4, 4, 4), (3, 3), 4, (2, 2), 1, 1),
+        ((1, 48, 48, 16), (3, 3), 16, (2, 2), 1, 1),
         ((1, 24, 24, 32), (3, 3), 32, (1, 1), 1, 1),
         ((1, 24, 24, 32), (3, 3), 32, (2, 2), 1, 1),
         ((1, 12, 12, 64), (3, 3), 64, (1, 1), 1, 1),
@@ -161,6 +162,7 @@ class TestDepthwiseConv2d_NHWC_HWOI_DSP(BasicDepthwiseConv2dTests):
         ((1, 6, 6, 128), (3, 3), 128, (2, 2), 1, 1),
         ((1, 3, 3, 256), (3, 3), 256, (1, 1), 1, 1),
 
+        # Asymmetric height and width
         ((1, 25, 5, 64), (3, 3), 64, (1, 1), 1, 1),
     )
     data_layout = tvm.testing.parameter("NHWC")