fixed algorithm. padding should be on both sides (#489)

apache · Sep 25, 2017 · c6a2045 · c6a2045
1 parent a3cbefd
commit c6a2045
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tutorials/optimize/opt_conv_cuda.py b/tutorials/optimize/opt_conv_cuda.py
@@ -36,10 +36,10 @@
 # Algorithm
 A = tvm.placeholder((in_size, in_size, in_channel, batch), name='A')
 W = tvm.placeholder((kernel, kernel, in_channel, out_channel), name='W')
-out_size = (in_size - kernel + pad) // stride + 1
+out_size = (in_size - kernel + 2*pad) // stride + 1
 # Pad input
 Apad = tvm.compute(
-    (in_size + pad, in_size + pad, in_channel, batch),
+    (in_size + 2*pad, in_size + 2*pad, in_channel, batch),
     lambda yy, xx, cc, nn: tvm.select(
         tvm.all(yy >= pad, yy - pad < in_size,
                 xx >= pad, xx - pad < in_size),