Skip to content
Snippets Groups Projects
Commit ca0fe22c authored by Wuwei Lin's avatar Wuwei Lin Committed by Tianqi Chen
Browse files

Add a testcase of dilated conv2d int8 (#2065)

parent de02a203
No related branches found
No related tags found
No related merge requests found
......@@ -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)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment