From ebb2dbd287a96082c847dbd6a7e33b1b1584c0a8 Mon Sep 17 00:00:00 2001 From: zhaoxinxin Date: Mon, 23 Nov 2020 20:15:43 +0800 Subject: [PATCH 1/2] modified: ge/host_kernels/strided_slice_kernel.cc --- ge/host_kernels/strided_slice_kernel.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ge/host_kernels/strided_slice_kernel.cc b/ge/host_kernels/strided_slice_kernel.cc index b76e5c6d..ddfb784b 100644 --- a/ge/host_kernels/strided_slice_kernel.cc +++ b/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); } } } From 3dd97e3498a1edae7e25d6b5f8ac387b47762375 Mon Sep 17 00:00:00 2001 From: zhaoxinxin Date: Mon, 23 Nov 2020 20:17:31 +0800 Subject: [PATCH 2/2] modified: ge/host_kernels/strided_slice_kernel.cc --- ge/host_kernels/strided_slice_kernel.cc | 1 - 1 file changed, 1 deletion(-) diff --git a/ge/host_kernels/strided_slice_kernel.cc b/ge/host_kernels/strided_slice_kernel.cc index ddfb784b..2fe74415 100644 --- a/ge/host_kernels/strided_slice_kernel.cc +++ b/ge/host_kernels/strided_slice_kernel.cc @@ -321,7 +321,6 @@ Status StridedSliceKernel::MaskCal(const size_t i, int64_t &begin_i, int64_t &en auto i_temp = static_cast(i); bool begin_mask_flag = (attr_value_map_.at(STRIDE_SLICE_ATTR_BEGIN_MASK) & (kMaskBitLeftUnit << i_temp)); bool end_mask_flag = (attr_value_map_.at(STRIDE_SLICE_ATTR_END_MASK) & (kMaskBitLeftUnit << i_temp)); - bool ellipsis_mask_flag = (attr_value_map_.at(STRIDE_SLICE_ATTR_ELLIPSIS_MASK) & (kMaskBitLeftUnit << i_temp)); bool shrink_mask_flag = (attr_value_map_.at(STRIDE_SLICE_ATTR_SHRINK_AXIS_MASK) & (kMaskBitLeftUnit << i_temp)); if (shrink_mask_flag) { begin_i = (begin_i < 0 ? (dim_i + begin_i) : begin_i);