IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Relax sizes and scales check for (#17811)

Browse Source
This commit is contained in:
Pawel Raasz 2023-05-31 17:33:48 +02:00 committed by GitHub
parent 9176f31085
commit b779dc3246
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 33 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/core/shape_inference/include/interpolate_shape_inference.hpp
View File

@ -73,7 +73,7 @@ inline void input_elements_num(const Node* const op,
size_t element_count, size_t element_count,
size_t exp_count) { size_t exp_count) {
NODE_VALIDATION_CHECK(op, NODE_VALIDATION_CHECK(op,
element_count == exp_count, element_count >= exp_count,
"The number of elements in the '", "The number of elements in the '",
input_name, input_name,
"' input does not match the number of axes ", "' input does not match the number of axes ",

32
src/core/tests/type_prop/interpolate.cpp
View File

@ -327,6 +327,38 @@ TEST(type_prop, interpolate_v4_use_scales_interval_shapes) {
ElementsAre(10, 11, ov::no_label, ov::no_label, ov::no_label)); ElementsAre(10, 11, ov::no_label, ov::no_label, ov::no_label));
} }
TEST(type_prop, interpolate_v4_target_shapes_gt_axes_number) {
const auto image = std::make_shared<op::Parameter>(element::f32, Shape{1, 3, 30, 60});
const auto target_shape = op::Constant::create<float>(element::i32, Shape{3}, {10, 12, 20});
const auto scales = op::Constant::create<float>(element::f32, Shape{1}, {0.3f});
const auto axes = op::Constant::create<int64_t>(element::i64, Shape{2}, {0, 3});
ov::op::util::InterpolateBase::InterpolateAttrs attrs;
attrs.shape_calculation_mode = ov::op::util::InterpolateBase::ShapeCalcMode::SIZES;
attrs.pads_begin = {0, 0, 0, 0};
attrs.pads_end = {0, 0, 0, 0};
auto interp = std::make_shared<op::v4::Interpolate>(image, target_shape, scales, axes, attrs);
EXPECT_EQ(interp->get_element_type(), element::f32);
EXPECT_EQ(interp->get_output_partial_shape(0), PartialShape({10, 3, 30, 12}));
}
TEST(type_prop, interpolate_v4_scales_gt_axes_number) {
const auto image = std::make_shared<op::Parameter>(element::f32, Shape{1, 3, 30, 60});
const auto target_shape = std::make_shared<op::Parameter>(element::i32, Shape{3});
const auto scales = op::Constant::create<float>(element::f32, Shape{3}, {0.2f, 0.2f, 0.3f});
const auto axes = op::Constant::create<int64_t>(element::i64, Shape{2}, {2, 3});
ov::op::util::InterpolateBase::InterpolateAttrs attrs;
attrs.shape_calculation_mode = ov::op::util::InterpolateBase::ShapeCalcMode::SCALES;
attrs.pads_begin = {0, 0, 0, 0};
attrs.pads_end = {0, 0, 0, 0};
auto interp = std::make_shared<op::v4::Interpolate>(image, target_shape, scales, axes, attrs);
EXPECT_EQ(interp->get_element_type(), element::f32);
EXPECT_EQ(interp->get_output_partial_shape(0), PartialShape({1, 3, 6, 12}));
}
TEST(type_prop, interpolate_v4_incorrect_mode) { TEST(type_prop, interpolate_v4_incorrect_mode) {
const auto image = std::make_shared<op::Parameter>(element::f32, Shape{1, 3, 30, 60}); const auto image = std::make_shared<op::Parameter>(element::f32, Shape{1, 3, 30, 60});
const auto target_shape = std::make_shared<op::Parameter>(element::i32, Shape{2}); const auto target_shape = std::make_shared<op::Parameter>(element::i32, Shape{2});