Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!49774 fix error log of slice 

Merge pull request !49774 from chengbin/fix_slice
This commit is contained in:
i-robot 2023-03-06 09:02:19 +00:00 committed by Gitee
parent 05b6f800b0 6a63f76b6e
commit c132b9c89d
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
mindspore/core/ops/slice.cc
View File

@ -49,7 +49,6 @@ namespace mindspore {
namespace ops {
namespace {
constexpr size_t kSliceInputNum = 3;
constexpr int64_t kDynamicOutValue = -2;
std::vector<int64_t> InferImplSliceFuncCalInputValue(const PrimitivePtr &primitive, const ValuePtr &input_value) {
std::vector<int64_t> tmp_input;
MS_EXCEPTION_IF_NULL(input_value);
@ -94,6 +93,14 @@ abstract::ShapePtr SliceInferShape(const PrimitivePtr &primitive, const std::vec
}
if (!IsValueKnown(input_size_value_ptr)) {
auto arg = input_args[kInputIndex2];
if (arg->isa<abstract::AbstractTensor>()) {
auto abs_tensor = arg->cast<abstract::AbstractTensorPtr>();
auto tensor_shape = abs_tensor->shape()->shape();
if (tensor_shape.size() != 1) {
MS_EXCEPTION(ValueError) << "For Slice, the shape of input|begin|size must be equal.";
}
}
out_shape = GetShapeValue(primitive, input_args[kInputIndex2]);
return std::make_shared<abstract::Shape>(out_shape);
}
@ -109,6 +116,11 @@ abstract::ShapePtr SliceInferShape(const PrimitivePtr &primitive, const std::vec
if (input_x_shape[i] < 0) {
continue;
}
if (input_size_value[i] < -1) {
MS_EXCEPTION(RuntimeError) << "For Slice, the value in size should not be less than -1, but got "
<< input_size_value[i];
}
if (input_begin_value[i] + input_size_value[i] > input_x_shape[i]) {
MS_EXCEPTION(ValueError) << "For Slice, the sum of begin_shape[" << i << "] and size_shape[" << i
<< "] must be no greater than input_x_shape[" << i << "].";