From 2864fe4e4d98294230f19a661c409f8dbfbafa68 Mon Sep 17 00:00:00 2001
From: Leyuan Wang <laurawly@gmail.com>
Date: Thu, 10 May 2018 18:18:30 -0700
Subject: [PATCH] dilation fixed for (1, 1) case (#477)
---
nnvm/python/nnvm/top/nn.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/nnvm/python/nnvm/top/nn.py b/nnvm/python/nnvm/top/nn.py
index ae9853db7..80e2fb698 100644
--- a/nnvm/python/nnvm/top/nn.py
+++ b/nnvm/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
--
GitLab