modified: ge/host_kernels/strided_slice_kernel.cc

ge/host_kernels/strided_slice_kernel.cc

@@ -298,7 +298,7 @@ void StridedSliceKernel::ExpandStrideWithEllipsisMask(const size_t x_dims_num,
       end_mask *= end_mask * (kMaskBitLeftUnit << (x_dims_num - orig_end_vec.size() - 1));
       attr_value_map_.at(STRIDE_SLICE_ATTR_END_MASK) = end_mask;
     }
-    for (auto i = 0; i < x_dims_num; ++i) {
+    for (size_t i = 0; i < x_dims_num; ++i) {
       bool ellipsis_mask_flag = attr_value_map_.at(STRIDE_SLICE_ATTR_ELLIPSIS_MASK) & (kMaskBitLeftUnit << i);
       if (ellipsis_mask_flag) {
         auto ellipsis_dim = i;
@@ -306,10 +306,10 @@ void StridedSliceKernel::ExpandStrideWithEllipsisMask(const size_t x_dims_num,
         orig_end_vec[i] = x_dims.at(i);
         orig_stride_vec[i] = 1;
         if (orig_begin_vec.size() < x_dims_num) {
-          for (auto j = 1; j < (x_dims_num - orig_begin_vec.size() + 1); ++j) {
+          for (size_t j = 1; j < (x_dims_num - orig_begin_vec.size() + 1); ++j) {
             orig_begin_vec.insert((orig_begin_vec.begin() + ellipsis_dim + j), 0);
             orig_end_vec.insert((orig_end_vec.begin() + ellipsis_dim + j), x_dims.at(ellipsis_dim +j));
-            orig_stride_vec.insert((orig_begin_vec.begin() + ellipsis_dim + j), 1);
+            orig_stride_vec.insert((orig_stride_vec.begin() + ellipsis_dim + j), 1);
           }
         }
       }