fix tests

openvinotoolkit · Nov 15, 2024 · 1202ab2 · 1202ab2
1 parent fa6b0ec
commit 1202ab2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...common/transformations/src/transformations/low_precision/mark_dequantization_subgraph.cpp b/...common/transformations/src/transformations/low_precision/mark_dequantization_subgraph.cpp
@@ -81,6 +81,8 @@ void swap_nodes(const PatternValueMap& pt_map,
         for (const auto& in : target_inputs) {
             in.replace_source_output(first_node->output(0));
         }
+        first_node->validate_and_infer_types();
+        second_node->validate_and_infer_types();
     }
 }
 
@@ -220,7 +222,5 @@ bool pass::MarkDequantizationAndDecompression::run_on_model(const std::shared_pt
     manager.register_pass<MarkDequantization>(m_precisions, m_fold_subtract_const, m_fold_multiply_const);
     manager.register_pass<ConstantFolding>();
     manager.register_pass<KeepConstsPrecision>(m_precisions, m_fold_subtract_const, m_fold_multiply_const);
-    manager.run_passes(m);
-
-    return false;
+    return manager.run_passes(m);
 }