diff --git a/src/common/transformations/src/transformations/op_conversions/convert_interpolate11_downgrade.cpp b/src/common/transformations/src/transformations/op_conversions/convert_interpolate11_downgrade.cpp index 5565b7c8162..38055b4a756 100644 --- a/src/common/transformations/src/transformations/op_conversions/convert_interpolate11_downgrade.cpp +++ b/src/common/transformations/src/transformations/op_conversions/convert_interpolate11_downgrade.cpp @@ -13,6 +13,7 @@ #include "openvino/op/constant.hpp" #include "openvino/op/interpolate.hpp" #include "openvino/op/shape_of.hpp" +#include "transformations/utils/utils.hpp" #include "utils.hpp" namespace { @@ -25,23 +26,23 @@ std::pair, ov::Output> make_v4_inputs( if (interpolate->get_input_size() == 3) { // broadcast dummy constant to the shape of axes - broadcast_shape = registry.make(interpolate->input_value(2)); + broadcast_shape = registry.add(ov::op::util::make_try_fold(interpolate->input_value(2))); } else { // broadcast dummy constant to the rank of data - broadcast_shape = registry.make(interpolate->input_value(0)); - broadcast_shape = registry.make(broadcast_shape); + broadcast_shape = registry.add(ov::op::util::make_try_fold(interpolate->input_value(0))); + broadcast_shape = registry.add(ov::op::util::make_try_fold(broadcast_shape)); } if (interpolate->get_attrs().shape_calculation_mode == ov::op::util::InterpolateBase::ShapeCalcMode::SCALES) { ret.second = interpolate->input_value(1); std::shared_ptr sizes_input = registry.make(ov::element::i32, ov::Shape{}, 1); - sizes_input = registry.make(sizes_input, broadcast_shape); + sizes_input = registry.add(ov::op::util::make_try_fold(sizes_input, broadcast_shape)); ret.first = sizes_input; } else { ret.first = interpolate->input_value(1); std::shared_ptr scales_input = registry.make(ov::element::f32, ov::Shape{}, 1.0f); - scales_input = registry.make(scales_input, broadcast_shape); + scales_input = registry.add(ov::op::util::make_try_fold(scales_input, broadcast_shape)); ret.second = scales_input; } diff --git a/src/common/transformations/tests/op_conversions/convert_interpolate11_downgrade_test.cpp b/src/common/transformations/tests/op_conversions/convert_interpolate11_downgrade_test.cpp index d385493e3e8..1f1d93eb0f5 100644 --- a/src/common/transformations/tests/op_conversions/convert_interpolate11_downgrade_test.cpp +++ b/src/common/transformations/tests/op_conversions/convert_interpolate11_downgrade_test.cpp @@ -74,23 +74,28 @@ std::shared_ptr create_v4_model(const bool with_axes, model_params.push_back(std::dynamic_pointer_cast(scales)); output_shape = ov::opset4::Constant::create(ov::element::i32, ov::Shape{}, {1}); if (with_axes) { - output_shape = - std::make_shared(output_shape, std::make_shared(axes)); - } else { - output_shape = std::make_shared( + output_shape = ov::op::util::make_try_fold( output_shape, - std::make_shared(std::make_shared(input))); + ov::op::util::make_try_fold(axes)); + } else { + output_shape = ov::op::util::make_try_fold( + output_shape, + ov::op::util::make_try_fold( + ov::op::util::make_try_fold(input))); } } else { output_shape = std::make_shared(ov::element::i32, ov::Shape{num_scales_or_sizes}); model_params.push_back(std::dynamic_pointer_cast(output_shape)); scales = ov::opset4::Constant::create(ov::element::f32, ov::Shape{}, {1.0f}); if (with_axes) { - scales = std::make_shared(scales, std::make_shared(axes)); - } else { - scales = std::make_shared( + scales = ov::op::util::make_try_fold( scales, - std::make_shared(std::make_shared(input))); + ov::op::util::make_try_fold(axes)); + } else { + scales = ov::op::util::make_try_fold( + scales, + ov::op::util::make_try_fold( + ov::op::util::make_try_fold(input))); } }