[LPT] NNCF GroupConvolution 5D on weights support: CPU plugin tests

openvinotoolkit · Mar 20, 2023 · 9ba8ac6 · 9ba8ac6
1 parent cf154df
commit 9ba8ac6
Showing 1 changed file with 117 additions and 0 deletions.
diff --git a/...red_tests_instances/low_precision_transformations/groupconvolution_qdq_transformation.cpp b/...red_tests_instances/low_precision_transformations/groupconvolution_qdq_transformation.cpp
@@ -370,6 +370,66 @@ const std::vector<LayerTestsDefinitions::GroupConvolutionQDqTransformationParam>
         true,
     },
 
+    // Actual:
+    //
+    // FQ
+    //  |FP32
+    //  |
+    // Convert    Convert   Constant  Constant
+    //  |U8        |U8       |I8       |I8
+    //  |          |         |         |
+    // Convert    Convert   Convert   Convert
+    //   \FP32    /FP32      \FP32    /FP32
+    //    \      /            \      /
+    //    Subtract  Constant  Subtract  Constant
+    //      \FP32   /FP32       \FP32   /FP32
+    //       \     /             \     /
+    //       Multiply           Multiply
+    //         \FP32           /FP32
+    //          \             /
+    //           \           /
+    //            \         /
+    //        GroupConvolution  Constant
+    //              \FP32       /FP32
+    //               \         /
+    //                 Multiply
+    //
+    // Transformed:
+    //
+    //  FQ        Constant Constant
+    //   \U8      /U8      / I8
+    //    \      /        /
+    //    Subtract     Subtract
+    //      \FP32      /FP32
+    //       \        /
+    //        \      /
+    //         \    /
+    //   GroupConvolution  Constant
+    //         \FP32       /FP32
+    //          \         /
+    //           Multiply
+    {
+        { 256ul, {{ 1, 1, 1, 1 }}, { -12.8f }, { 12.7f }, { 0.f }, { 255.f }, ngraph::element::f32 },
+        { ngraph::element::u8, false },
+        {
+            { ngraph::element::f32, false },
+            { {128.f}, ngraph::element::f32, {}, false, 1ul, ngraph::element::u8, true },
+            { {0.1f}, ngraph::element::f32, {}, false }
+        },
+        { std::vector<float>(4, 15.f), ngraph::element::i8, {2, 1, 2, 1, 1} },
+        {},
+        {},
+        {
+            { ngraph::element::f32, false },
+            { {126.f, 127.f}, ngraph::element::f32, {2, 1, 1, 1, 1}, false, 1ul, ngraph::element::i8, true },
+            { {0.1f, 0.2f}, ngraph::element::f32, {2, 1, 1, 1, 1}, false }
+        },
+        {},
+        "output_original",
+        "FP32",
+        true,
+    },
+
     // Actual:
     //
     // FQ
@@ -427,6 +487,63 @@ const std::vector<LayerTestsDefinitions::GroupConvolutionQDqTransformationParam>
         false,
     },
 
+    // Actual:
+    //
+    // FQ
+    //  |FP32
+    //  |
+    // Convert    Convert
+    //  |U8        |U8
+    //  |          |
+    // Convert    Convert   Constant
+    //   \FP32    /FP32      \U8
+    //    \      /            \
+    //    Subtract  Constant  Convert   Constant
+    //      \FP32   /FP32       \FP32   /FP32
+    //       \     /             \     /
+    //       Multiply           Multiply
+    //         \FP32           /FP32
+    //          \             /
+    //           \           /
+    //            \         /
+    //          GroupConvolution
+    //
+    // Transformed:
+    //
+    //  FQ        Constant
+    //   \U8      /U8
+    //    \      /
+    //    Subtract
+    //      \FP32
+    //       \        Constant
+    //        \       /I8
+    //         \     /
+    //   GroupConvolution   Constant
+    //           \FP32      /FP32
+    //            \        /
+    //             Multiply
+    {
+        { 256ul, {{ 1, 1, 1, 1 }}, { -12.8f }, { 12.7f }, { 0.f }, { 255.f }, ngraph::element::f32 },
+        { ngraph::element::u8, false },
+        {
+            { ngraph::element::f32, false },
+            { {128.f}, ngraph::element::f32, {}, false, 1ul, ngraph::element::u8, true },
+            { {0.1f}, ngraph::element::f32, {}, false }
+        },
+        { std::vector<float>(4, 15.f), ngraph::element::i8, {2, 1, 2, 1, 1} },
+        {},
+        {},
+        {
+            { ngraph::element::f32, false },
+            {},
+            { {0.1f, 0.2f}, ngraph::element::f32, {2, 1, 1, 1, 1}, false }
+        },
+        {},
+        "output_original",
+        "U8",
+        false,
+    },
+
     // Actual:
     //
     // FQ