added missing definition GetPaddingSizeFull

ChaiBapchya · May 4, 2020 · 62be87b · 62be87b
1 parent 5bf9c76
commit 62be87b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/operator/nn/mkldnn/mkldnn_pooling-inl.h b/src/operator/nn/mkldnn/mkldnn_pooling-inl.h
@@ -87,6 +87,14 @@ class MKLDNNPoolingBwd {
   const mkldnn::pooling_backward::primitive_desc &GetPd();
 };
 
+inline int GetPaddingSizeFull(dim_t x, int padl, int padr, int k, int s) {
+  if ((x + padl + padr - k) % s != 0) {
+    return (padr + s - ((x + padl + padr - k) % s));
+  } else {
+    return padr;
+  }
+}
+
 inline bool SupportMKLDNNPooling(const PoolingParam &param) {
   return (param.kernel.ndim() == 1 || param.kernel.ndim() == 2 ||
           param.kernel.ndim() == 3) &&