test fix: interval changes #2

eshoguli · Mar 22, 2023 · 682db9d · 682db9d
1 parent 5ee47bc
commit 682db9d
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/...hared_tests_instances/low_precision_transformations/fuse_fake_quantize_transformation.cpp b/...hared_tests_instances/low_precision_transformations/fuse_fake_quantize_transformation.cpp
@@ -33,8 +33,8 @@ const std::vector<FuseFakeQuantizeTransformationTestValues> testValues = {
         LayerTestsUtils::LayerTransformationParamsNGraphFactory::createParamsU8I8(),
         {
             ngraph::element::f32,
-            { 256ul, {}, { -12.8f }, { 12.7f }, { -12.8f }, { 12.7f }, ngraph::element::f32 },
-            { 256ul, {}, { -12.8f }, { 12.7f }, { -12.8f }, { 12.7f }, ngraph::element::f32 }
+            { 256ul, {}, { -12.8f * 2.f }, { 12.7f * 2.f }, { -12.8f * 2.f }, { 12.7f * 2.f }, ngraph::element::f32 },
+            { 256ul, {}, { -12.8f * 2.f }, { 12.7f * 2.f }, { -12.8f * 2.f }, { 12.7f * 2.f }, ngraph::element::f32 }
         },
         {
             { "fakeQuantize1", "fakeQuantize2" }, // not fused
@@ -48,7 +48,7 @@ const std::vector<FuseFakeQuantizeTransformationTestValues> testValues = {
         {
             ngraph::element::f32,
             { 256ul, {}, { 0.f }, { 25.5f }, { 0.f }, { 25.5f }, ngraph::element::f32 },
-            { 256ul, {}, { 0.f }, { 25.5f / 2.f }, { 0.f }, { 25.5f / 2.f }, ngraph::element::f32 }
+            { 256ul, {}, { 0.f }, { 25.5f / 1.1f }, { 0.f }, { 25.5f / 1.1f }, ngraph::element::f32 }
         },
         {
             { "fakeQuantize1", "fakeQuantize2" }, // not fused

diff --git a/...plugin/shared/include/low_precision_transformations/fuse_fake_quantize_transformation.hpp b/...plugin/shared/include/low_precision_transformations/fuse_fake_quantize_transformation.hpp
@@ -47,6 +47,8 @@ class FuseFakeQuantizeTransformation
 public:
     static std::string getTestCaseName(const testing::TestParamInfo<FuseFakeQuantizeTransformationParams>& obj);
 
+    virtual InferenceEngine::Blob::Ptr GenerateInput(const InferenceEngine::InputInfo& inputInfo) const;
+
 protected:
     void SetUp() override;
 };

diff --git a/...nal/plugin/shared/src/low_precision_transformations/fuse_fake_quantize_transformation.cpp b/...nal/plugin/shared/src/low_precision_transformations/fuse_fake_quantize_transformation.cpp
@@ -27,6 +27,10 @@ std::string FuseFakeQuantizeTransformation::getTestCaseName(const testing::TestP
     return result.str();
 }
 
+InferenceEngine::Blob::Ptr FuseFakeQuantizeTransformation::GenerateInput(const InferenceEngine::InputInfo& info) const {
+    return FuncTestUtils::createAndFillBlob(info.getTensorDesc(), 20, 0, 1, 1);
+}
+
 void FuseFakeQuantizeTransformation::SetUp() {
     FuseFakeQuantizeTransformationTestValues testValues;
     std::tie(targetDevice, testValues) = this->GetParam();