rebase fix

eshoguli · Sep 13, 2023 · 7b1a9cc · 7b1a9cc
1 parent 69190f8
commit 7b1a9cc
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 12 deletions.
diff --git a/...al/shared_tests_instances/low_precision_transformations/batch_to_space_transformation.cpp b/...al/shared_tests_instances/low_precision_transformations/batch_to_space_transformation.cpp
@@ -21,7 +21,7 @@ const std::vector<BatchToSpaceTransformationParam> params = {
         { 1, 1, 2, 2 }, { 0, 0, 0, 0 }, { 0, 0, 0, 1 },
         { 256ul, ngraph::Shape{ 1, 1, 1, 1 }, { 0.f }, { 2.55f }, { 0.f }, { 2.55f } },
         "batch_to_space",
-        "U8"
+        "u8"
     },
     {
         { 4, 3, 50, 86 },
@@ -35,7 +35,7 @@ const std::vector<BatchToSpaceTransformationParam> params = {
             { 255.f, 255.f/2.f, 255.f/3.f },
         },
         "batch_to_space",
-        "FP32"
+        "f32"
     }
 };
 

diff --git a/...al/shared_tests_instances/low_precision_transformations/space_to_batch_transformation.cpp b/...al/shared_tests_instances/low_precision_transformations/space_to_batch_transformation.cpp
@@ -21,7 +21,7 @@ const std::vector<SpaceToBatchTransformationParam> params = {
         { 1, 1, 2, 2 }, { 0, 0, 2, 2 }, { 0, 0, 2, 3 },
         { 256ul, ngraph::Shape{ 1, 1, 1, 1 }, { 0.f }, { 2.55f }, { 0.f }, { 2.55f } },
         "space_to_batch",
-        "U8"
+        "u8"
     },
     {
         {1, 3, 100, 171},
@@ -35,7 +35,7 @@ const std::vector<SpaceToBatchTransformationParam> params = {
             { 255.f, 255.f/2.f, 255.f/3.f },
         },
         "space_to_batch",
-        "FP32"
+        "f32"
     }
 };
 

diff --git a/...ctional/plugin/shared/src/low_precision_transformations/batch_to_space_transformation.cpp b/...ctional/plugin/shared/src/low_precision_transformations/batch_to_space_transformation.cpp
@@ -42,11 +42,13 @@ void BatchToSpaceTransformation::Run() {
     LayerTestsCommon::Run();
 
     const auto params = std::get<2>(GetParam());
-    auto actual_type = getRuntimePrecisionByType(params.layer_type);
-    const auto expected_type = params.expected_kernel_type;
-    if ((expected_type == "FP32") && (actual_type == "FP16")) {
-        actual_type = "FP32";
+    auto expected_type = params.expected_kernel_type;
+    const auto input_type = std::get<0>(GetParam());
+    if ((expected_type == "f32") && (input_type == ov::element::f16)) {
+        expected_type = "f16";
     }
+
+    const auto actual_type = getRuntimePrecisionByType(params.layer_type);
     EXPECT_EQ(actual_type, expected_type);
 }
 

diff --git a/...ctional/plugin/shared/src/low_precision_transformations/space_to_batch_transformation.cpp b/...ctional/plugin/shared/src/low_precision_transformations/space_to_batch_transformation.cpp
@@ -42,11 +42,13 @@ void SpaceToBatchTransformation::Run() {
     LayerTestsCommon::Run();
 
     const auto params = std::get<2>(GetParam());
-    auto actual_type = getRuntimePrecisionByType(params.layer_type);
-    const auto expected_type = params.expected_kernel_type;
-    if ((expected_type == "FP32") && (actual_type == "FP16")) {
-        actual_type = "FP32";
+    auto expected_type = params.expected_kernel_type;
+    const auto input_type = std::get<0>(GetParam());
+    if ((expected_type == "f32") && (input_type == ov::element::f16)) {
+        expected_type = "f16";
     }
+
+    const auto actual_type = getRuntimePrecisionByType(params.layer_type);
     EXPECT_EQ(actual_type, expected_type);
 }