Browse Source

!391 Bugfix: fix stridedslice code style

From: @hugo1
Reviewed-by: @xchu42,@wqtshg
Signed-off-by: @wqtshg
tags/v1.1.0
mindspore-ci-bot Gitee 3 years ago
parent
commit
53d6e771a8
1 changed files with 3 additions and 4 deletions
  1. +3
    -4
      ge/host_kernels/strided_slice_kernel.cc

+ 3
- 4
ge/host_kernels/strided_slice_kernel.cc View File

@@ -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);
}
}
}
@@ -321,7 +321,6 @@ Status StridedSliceKernel::MaskCal(const size_t i, int64_t &begin_i, int64_t &en
auto i_temp = static_cast<uint32_t>(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);


Loading…
Cancel
Save