quick fix

eshoguli · Sep 14, 2023 · 1c9542d · 1c9542d
1 parent 67a4d9c
commit 1c9542d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/src/common/low_precision_transformations/tests/multiply_partial_transformation.cpp b/src/common/low_precision_transformations/tests/multiply_partial_transformation.cpp
@@ -13,7 +13,7 @@
 #include <utility>
 #include <transformations/utils/utils.hpp>
 #include <transformations/init_node_info.hpp>
-#include "low_precision/multiply.hpp"
+#include "low_precision/multiply_partial.hpp"
 #include "lpt_ngraph_functions/common/dequantization_operations.hpp"
 
 #include "common_test_utils/ov_test_utils.hpp"
@@ -54,8 +54,9 @@ class MultiplyPartialTransformation : public LayerTransformation, public testing
         const MultiplyPartialTransformationTestValues testParams = std::get<1>(GetParam());
 
         actualFunction = MultiplyPartialFunction::get(precision, testParams.actual);
+
         SimpleLowPrecisionTransformer transform;
-        transform.add<ngraph::pass::low_precision::MultiplyTransformation, ov::op::v1::Multiply>(testParams.transformationParams);
+        transform.add<ngraph::pass::low_precision::MultiplyPartialTransformation, ov::op::v1::Multiply>(testParams.transformationParams);
         transform.transform(actualFunction);
 
         referenceFunction = MultiplyPartialFunction::get(precision, testParams.expected);

diff --git a/src/common/low_precision_transformations/tests/multiply_transformation.cpp b/src/common/low_precision_transformations/tests/multiply_transformation.cpp
@@ -20,8 +20,6 @@
 #include "simple_low_precision_transformer.hpp"
 #include "lpt_ngraph_functions/multiply_function.hpp"
 
-#include "ngraph/pass/serialize.hpp"
-
 namespace {
 using namespace testing;
 using namespace ov;