Skip to content

Commit

Permalink
[LPT] [NPU] Multiply support #2
Browse files Browse the repository at this point in the history
  • Loading branch information
eshoguli committed Sep 28, 2023
1 parent 2d916d5 commit 5d5b09c
Showing 1 changed file with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -14,14 +14,13 @@
#include "transformations/utils/utils.hpp"
#include "transformations/init_node_info.hpp"
#include "low_precision/multiply.hpp"
#include "low_precision/multiply_to_group_convolution.hpp"
#include "lpt_ngraph_functions/common/dequantization_operations.hpp"

#include "common_test_utils/ov_test_utils.hpp"
#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;
Expand Down Expand Up @@ -122,8 +121,9 @@ class MultiplyTransformation : public LayerTransformation, public testing::WithP

actualFunction = MultiplyFunction::get(model_precision, to_multiply_values(testParams.actual));

SimpleLowPrecisionTransformer transform;
SimpleLowPrecisionTransformer transform({}, {}, AttributeParameters(), true);
transform.add<ov::pass::low_precision::MultiplyTransformation, ov::op::v1::Multiply>(testParams.transformationParams);
transform.cleanup->get_pass_config()->disable<ov::pass::low_precision::MultiplyToGroupConvolutionTransformation>();
transform.transform(actualFunction);

referenceFunction = MultiplyFunction::get(model_precision, to_multiply_values(testParams.expected));
Expand Down

0 comments on commit 5d5b09c

Please sign in to comment.