small improvm

openvinotoolkit · Dec 16, 2024 · 765d227 · 765d227
1 parent a6efe14
commit 765d227
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/common/transformations/tests/utils/convert_precision.cpp b/src/common/transformations/tests/utils/convert_precision.cpp
@@ -2001,7 +2001,7 @@ static std::shared_ptr<ov::Model> make_then_body(bool ref) {
     auto add_conv = std::make_shared<opset10::Convert>(add_const_1, element::f32);
     auto add_1 = std::make_shared<opset10::Multiply>(mul, add_conv);
 
-    auto res_conv = ref ? std::make_shared<opset10::Convert>(add_1, element::f32)->output(0) : add_1;
+    auto res_conv = ref ? std::make_shared<opset10::Convert>(add_1, element::f16)->output(0) : add_1->output(0);
 
     auto then_res = std::make_shared<opset10::Result>(res_conv);