mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!34788 change nmsmask checktype order 

Merge pull request !34788 from TuDouNi/pass_support_dynamic
This commit is contained in:
i-robot 2022-05-25 01:46:25 +00:00 committed by Gitee
parent d31dc6c97e dc3f7ae19f
commit 8be4f542e7
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/core/ops/nms_with_mask.cc
View File

@ -64,7 +64,8 @@ abstract::TupleShapePtr NMSWithMaskInferShape(const PrimitivePtr &primitive,
(void)CheckAndConvertUtils::CheckValue<size_t>("shape of bboxes", bboxes_shape.size(), kEqual, kBboxesShapeSize,
op_name);
if (bboxes_shape[1] != kBboxesShapeIn2ndDimNormal && bboxes_shape[1] != kBboxesShapeIn2ndDimAscendAfterPad) {
MS_EXCEPTION(ValueError) << "The 2nd dim in shape of bboxes should equal to 5 or 8, but got " << bboxes_shape[1];
MS_EXCEPTION(ValueError) << " For " << op_name
<< ", the 2nd dim in shape of bboxes should equal to 5 or 8, but got " << bboxes_shape[1];
}
// for ascend
@ -121,8 +122,9 @@ AbstractBasePtr NMSWithMaskInfer(const abstract::AnalysisEnginePtr &, const Prim
MS_EXCEPTION_IF_NULL(item);
}
return abstract::MakeAbstract(NMSWithMaskInferShape(primitive, input_args),
NMSWithMaskInferType(primitive, input_args));
auto infer_type = NMSWithMaskInferType(primitive, input_args);
auto infer_shape = NMSWithMaskInferShape(primitive, input_args);
return abstract::MakeAbstract(infer_shape, infer_type);
}
REGISTER_PRIMITIVE_EVAL_IMPL(NMSWithMask, prim::kPrimNMSWithMask, NMSWithMaskInfer, nullptr, true);
} // namespace ops