|
|
|
@ -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);
|
|
|
|
|