fix same issue in pooling

apache · Nov 7, 2019 · 009b141 · 009b141
1 parent 7a391af
commit 009b141
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/operator/nn/mkldnn/mkldnn_pooling.cc b/src/operator/nn/mkldnn/mkldnn_pooling.cc
@@ -127,7 +127,7 @@ mkldnn::algorithm GetMKLDNNPoolAlgo(const PoolingParam &param) {
   }
 }
 
-static inline int GetPaddingSizeFull(int x, int padl, int padr, int k, int s) {
+static 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 {