dilation fixed for (1, 1) case (dmlc#477)

padmaramesh · May 11, 2018 · 6eb04ae · 6eb04ae
1 parent 5363259
commit 6eb04ae
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/nnvm/top/nn.py b/python/nnvm/top/nn.py
@@ -88,6 +88,8 @@ def compute_conv2d(attrs, inputs, _):
     (dilation_h, dilation_w) = dilation
     if dilation_h < 1 or dilation_w < 1:
         raise ValueError("dilation should be positive value")
+    elif dilation == (1, 1):
+        kernel = inputs[1]
     elif layout == "NCHW":
         kernel = topi.nn.dilate(inputs[1], [1, 1, dilation_h, dilation_w])
     else: #layout == NHWC