Support pad for GeometryPoolGrad

patrickket · Jun 22, 2021 · 2733909 · 2733909
1 parent f6422c3
commit 2733909
Showing 1 changed file with 18 additions and 25 deletions.
diff --git a/source/geometry/GeometryPoolGrad.cpp b/source/geometry/GeometryPoolGrad.cpp
@@ -47,19 +47,16 @@ class GeometryPoolGrad : public GeometryComputer {
             stride_h = ih;
             pad_w    = 0;
             pad_h    = 0;
-        }
-
-        if (parameter->padType() == PoolPadType_SAME) {
-            int pad_w_total = (ow - 1) * stride_w + kernel_w - iw;
-            int pad_h_total = (oh - 1) * stride_h + kernel_h - ih;
-            pad_w           = pad_w_total > 0 ? pad_w_total / 2 : 0;
-            pad_h           = pad_h_total > 0 ? pad_h_total / 2 : 0;
-        } else if (parameter->padType() == PoolPadType_VALID) {
-            pad_w = 0;
-            pad_h = 0;
         } else {
-            MNN_PRINT("Pool padtype not supported!\n");
-            return false;
+            if (parameter->padType() == PoolPadType_SAME) {
+                int pad_w_total = (ow - 1) * stride_w + kernel_w - iw;
+                int pad_h_total = (oh - 1) * stride_h + kernel_h - ih;
+                pad_w           = pad_w_total > 0 ? pad_w_total / 2 : 0;
+                pad_h           = pad_h_total > 0 ? pad_h_total / 2 : 0;
+            } else if (parameter->padType() == PoolPadType_VALID) {
+                pad_w = 0;
+                pad_h = 0;
+            }
         }
 
         std::vector<std::shared_ptr<Tensor>> originSplit;
@@ -304,21 +301,17 @@ class GeometryPoolGrad : public GeometryComputer {
             stride_h = ih;
             pad_w    = 0;
             pad_h    = 0;
-        }
-
-        if (parameter->padType() == PoolPadType_SAME) {
-            int pad_w_total = (ow - 1) * stride_w + kernel_w - iw;
-            int pad_h_total = (oh - 1) * stride_h + kernel_h - ih;
-            pad_w           = pad_w_total > 0 ? pad_w_total / 2 : 0;
-            pad_h           = pad_h_total > 0 ? pad_h_total / 2 : 0;
-        } else if (parameter->padType() == PoolPadType_VALID) {
-            pad_w = 0;
-            pad_h = 0;
         } else {
-            MNN_PRINT("Pool padtype not supported!\n");
-            return false;
+            if (parameter->padType() == PoolPadType_SAME) {
+                int pad_w_total = (ow - 1) * stride_w + kernel_w - iw;
+                int pad_h_total = (oh - 1) * stride_h + kernel_h - ih;
+                pad_w           = pad_w_total > 0 ? pad_w_total / 2 : 0;
+                pad_h           = pad_h_total > 0 ? pad_h_total / 2 : 0;
+            } else if (parameter->padType() == PoolPadType_VALID) {
+                pad_w = 0;
+                pad_h = 0;
+            }
         }
-
         std::shared_ptr<Tensor> inpDifTrans;
 
         inpDifTrans.reset(new Tensor);