Add a testcase of dilated conv2d int8 (#2065)

ca0fe22c · Wuwei Lin · Tianqi Chen · de02a203 · ca0fe22c
Commit ca0fe22c authored Nov 06, 2018 by Wuwei Lin Committed by Tianqi Chen Nov 05, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

topi/tests/python/test_topi_conv2d_int8.py
+3 -0

No files found.
--- a/topi/tests/python/test_topi_conv2d_int8.py
+++ b/topi/tests/python/test_topi_conv2d_int8.py
@@ -119,6 +119,9 @@ def test_conv2d_nchw():
        verify_conv2d_NCHWc_int8(1, 64, 56, 64, 3, 1, 1, add_bias=True)
        verify_conv2d_NCHWc_int8(1, 64, 56, 64, 3, 1, 1, add_bias=True, add_relu=True)

+        # dilation = 2
+        verify_conv2d_NCHWc_int8(1, 64, 56, 64, 3, 1, 1, dilation=2)
+
        # batch size
        verify_conv2d_NCHWc_int8(4, 64, 56, 64, 3, 1, 1)
        verify_conv2d_NCHWc_int8(9, 64, 56, 64, 3, 1, 1)