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

fix scale_and_translate dynamic shape

This commit is contained in:
yeyunpeng2020 2023-03-03 16:52:47 +08:00
parent a7f81dcf82
commit 37af9c8ea4
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mindspore/core/ops/scale_and_translate.cc
View File

@ -41,6 +41,13 @@ abstract::ShapePtr ScaleAndTranslateInferShape(const PrimitivePtr &primitive,
IsDynamicRank(translation_shape)) {
return std::make_shared<abstract::Shape>(ShapeVector({abstract::Shape::kShapeRankAny}));
}
// support dynamic shape
if (IsDynamicShape(images_shape) || IsDynamicShape(size_shape) || IsDynamicShape(scale_shape) ||
IsDynamicShape(translation_shape)) {
return std::make_shared<abstract::Shape>(
ShapeVector({abstract::Shape::kShapeDimAny, abstract::Shape::kShapeDimAny, abstract::Shape::kShapeDimAny,
abstract::Shape::kShapeDimAny}));
}
const int64_t kShapeSize = 1;
const int64_t kElementsNumber = 2;