Skip to content

T
tic
Commit ca0fe22c by Wuwei Lin Committed by Tianqi Chen
Options

Add a testcase of dilated conv2d int8 (#2065)

parent de02a203
Showing with 3 additions and 0 deletions
topi/tests/python/test_topi_conv2d_int8.py
...@@ -119,6 +119,9 @@ def test_conv2d_nchw(): ...@@ -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)
verify_conv2d_NCHWc_int8(1, 64, 56, 64, 3, 1, 1, add_bias=True, add_relu=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 # batch size
verify_conv2d_NCHWc_int8(4, 64, 56, 64, 3, 1, 1) verify_conv2d_NCHWc_int8(4, 64, 56, 64, 3, 1, 1)
verify_conv2d_NCHWc_int8(9, 64, 56, 64, 3, 1, 1) verify_conv2d_NCHWc_int8(9, 64, 56, 64, 3, 1, 1)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment