[TOPI] Add dilation argument to conv2d and depthwise_conv2d (apache#1970

)
tqchen · Oct 31, 2018 · 0fc5812 · 0fc5812
1 parent 9ab4128
commit 0fc5812
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/python/integration/test_benchmark_topi_conv2d.py b/tests/python/integration/test_benchmark_topi_conv2d.py
@@ -33,6 +33,7 @@ def run_cpu_conv2d(env, remote, key, batch_size, wl, profile=True):
         res_conv = topi.nn.conv2d(
             data, kernel, padding=(wl.hpad, wl.wpad),
             strides=(wl.hstride, wl.wstride),
+            dilation=(1, 1),
             out_dtype="int32")
         res = topi.right_shift(res_conv, 8)
         res = my_clip(res, 0, 127)