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

Explicit scales for all axes in interpolate op test (#16404)

Browse Source
This commit is contained in:
Tomasz Dołbniak 2023-03-20 17:50:47 +01:00 committed by GitHub
parent 9c69e2f694
commit d9c70dbce3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -81,8 +81,7 @@ 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, {2}, {2.0, 2.0});
const auto axes = op::v0::Constant::create(element::i32, {2}, {2, 3});
const auto scales = op::v0::Constant::create(element::f32, {4}, {1.0, 1.0, 2.0, 2.0});
op::v11::Interpolate::InterpolateAttrs attrs;
attrs.mode = op::v11::Interpolate::InterpolateMode::BILINEAR_PILLOW;
@ -94,7 +93,7 @@ TEST(attributes, interpolate_op11) {
attrs.antialias = true;
attrs.cube_coeff = -0.75;
auto interpolate = make_shared<opset11::Interpolate>(img, scales, axes, attrs);
auto interpolate = make_shared<opset11::Interpolate>(img, scales, attrs);
NodeBuilder builder(interpolate, {img, scales});
auto g_interpolate = ov::as_type_ptr<opset11::Interpolate>(builder.create());