modified: ge/host_kernels/strided_slice_kernel.cc

ge/host_kernels/strided_slice_kernel.cc

@@ -305,10 +305,12 @@ void StridedSliceKernel::ExpandStrideWithEllipsisMask(const size_t x_dims_num,
         orig_begin_vec[i] = 0;
         orig_end_vec[i] = x_dims.at(i);
         orig_stride_vec[i] = 1;
-        if (auto j = 0; 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);
+        if (orig_begin_vec.size() < x_dims_num) {
+          for (auto j = 0; 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);
+          }
         }
       }
     }