[MKLDNN] use dim_t instead of int in slice/transpose operators (#16737)

* use dim_t instead of int

* fix same issue in pooling

* rebase code

* trigger CI

Author: wuxun-zhang

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 {

src/operator/nn/mkldnn/mkldnn_slice.cc

@@ -41,7 +41,7 @@ MKLDNNSliceFwd::MKLDNNSliceFwd(const SliceParam &param,
   mkldnn::memory::dims dims(N);
   mkldnn::memory::dims offsets(N);
   for (int i = 0; i < N; ++i) {
-    int s = 0;
+    dim_t s = 0;
     if (i < param.begin.ndim() &&  param.begin[i]) {
       s = *param.begin[i];
       if (s < 0) s += ishape[i];

src/operator/nn/mkldnn/mkldnn_transpose.cc

@@ -73,7 +73,7 @@ class MKLDNNTransposeForward {
 
     mkldnn_dims_t strides;
     mkldnn_dims_t sh;
-    unsigned int total_stride = 1;
+    dim_t total_stride = 1;
     for (int i = data_ndim - 1; i >= 0; i--) {
       sh[i] = shape[i];
       strides[axes[i]] = total_stride;