From 2e3f8e7498079d1f5346af1d184c9d80b2071417 Mon Sep 17 00:00:00 2001
From: ziheng <ziheng@apache.org>
Date: Mon, 27 Nov 2017 14:00:42 -0800
Subject: [PATCH] [TOPI] Fix for pooling (#673)
---
topi/python/topi/nn/pooling.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/topi/python/topi/nn/pooling.py b/topi/python/topi/nn/pooling.py
index 955f72422..99b15e18e 100644
--- a/topi/python/topi/nn/pooling.py
+++ b/topi/python/topi/nn/pooling.py
@@ -81,14 +81,14 @@ def pool(data, kernel, stride, padding, pool_type, ceil_mode=False):
pad_top, pad_left, pad_down, pad_right = get_pad_tuple(
padding, (kernel_height, kernel_width))
- pad_before = [0, 0, pad_top, pad_left]
- pad_after = [0, 0, pad_down, pad_right]
-
if ceil_mode:
# Additional padding to ensure we do ceil instead of floor when divide stride.
pad_down += stride_height -1
pad_right += stride_width - 1
+ pad_before = [0, 0, pad_top, pad_left]
+ pad_after = [0, 0, pad_down, pad_right]
+
out_height = util.simplify((height - kernel_height + pad_top + pad_down) // stride_height + 1)
out_width = util.simplify((width - kernel_width + pad_left + pad_right) // stride_width + 1)
--
GitLab