From 68891014155cca76fe80cc5119e95c25be375cb9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Tomasz=20Do=C5=82bniak?= Date: Fri, 17 Mar 2023 18:49:46 +0100 Subject: [PATCH] Axes bugfix (#16365) --- src/core/tests/visitors/op/interpolate.cpp | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/core/tests/visitors/op/interpolate.cpp b/src/core/tests/visitors/op/interpolate.cpp index 23e0178e5a8..fe375514439 100644 --- a/src/core/tests/visitors/op/interpolate.cpp +++ b/src/core/tests/visitors/op/interpolate.cpp @@ -81,7 +81,8 @@ TEST(attributes, interpolate_op4) { TEST(attributes, interpolate_op11) { NodeBuilder::get_ops().register_factory(); const auto img = make_shared(element::f32, Shape{1, 3, 32, 32}); - const auto scales = op::v0::Constant::create(element::f32, {1}, {1.0}); + const auto scales = op::v0::Constant::create(element::f32, {2}, {2.0, 2.0}); + const auto axes = op::v0::Constant::create(element::i32, {2}, {2, 3}); op::v11::Interpolate::InterpolateAttrs attrs; attrs.mode = op::v11::Interpolate::InterpolateMode::BILINEAR_PILLOW; @@ -93,7 +94,7 @@ TEST(attributes, interpolate_op11) { attrs.antialias = true; attrs.cube_coeff = -0.75; - auto interpolate = make_shared(img, scales, attrs); + auto interpolate = make_shared(img, scales, axes, attrs); NodeBuilder builder(interpolate, {img, scales}); auto g_interpolate = ov::as_type_ptr(builder.create());