|
|
@@ -221,7 +221,9 @@ Status StridedSliceKernel::InitParamWithAttrs(const std::vector<ConstGeTensorPtr |
|
|
|
// handle new_axis_mask |
|
|
|
ExpandDimsWithNewAxis(begin_tensor, x_dims_num, x_dims); |
|
|
|
|
|
|
|
vector<int64_t> orig_begin_vec, orig_end_vec, orig_stride_vec; |
|
|
|
vector<int64_t> orig_begin_vec; |
|
|
|
vector<int64_t> orig_end_vec; |
|
|
|
vector<int64_t> orig_stride_vec; |
|
|
|
GetOriginStrideVec(input, orig_begin_vec, orig_end_vec, orig_stride_vec); |
|
|
|
// calculate begin_mask & end_mask by ellipsis_mask |
|
|
|
ExpandStrideWithEllipsisMask(x_dims_num, x_dims, orig_begin_vec, orig_end_vec, orig_stride_vec); |
|
|
|