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

Axes bugfix (#16365)

Browse Source
This commit is contained in:
Tomasz Dołbniak
2023-03-17 18:49:46 +01:00
committed by GitHub
parent d189df169c
commit 6889101415
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/core/tests/visitors/op/interpolate.cpp
View File

@@ -81,7 +81,8 @@ TEST(attributes, interpolate_op4) {
TEST(attributes, interpolate_op11) {
NodeBuilder::get_ops().register_factory<opset11::Interpolate>();
const auto img = make_shared<op::Parameter>(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<opset11::Interpolate>(img, scales, attrs);
auto interpolate = make_shared<opset11::Interpolate>(img, scales, axes, attrs);
NodeBuilder builder(interpolate, {img, scales});
auto g_interpolate = ov::as_type_ptr<opset11::Interpolate>(builder.create());