commit 6a2845317f9f9e5161f3afab2b54dac494f57f89
author Sadik Armagan <sadik.armagan@arm.com> Fri Mar 12 09:12:29 2021 +0000
committer Sadik Armagan <sadik.armagan@arm.com> Fri Mar 12 09:12:29 2021 +0000
tree fbaaae4f162b80adc5998d2b77641a80d0bed59e
parent 89cbb3a914d76843630f3576108e309cf085ef80

IVGCVSW-5767 'Regression on SqueezeNet v1.1'

* MEAN operator second input data type is different than Reduction operators
* Use existing ConvertMean() function for MEAN operator

Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I211cb77515d88e54ded9646eebe3e901d437827c

1.3/HalPolicy.cpp

diff --git a/1.3/HalPolicy.cpp b/1.3/HalPolicy.cpp
index 9a9f2ac..d58ac6c 100644
--- a/1.3/HalPolicy.cpp
+++ b/1.3/HalPolicy.cpp
@@ -98,7 +98,7 @@
         case V1_3::OperationType::MAXIMUM:
             return ConvertMaximum(operation, model, data);
         case V1_3::OperationType::MEAN:
-            return ConvertReduce(operation, model, data, ReduceOperation::Mean);
+            return ConvertMean(operation, model, data);
         case V1_3::OperationType::MINIMUM:
             return ConvertMinimum(operation, model, data);
         case V1_3::OperationType::MUL:
@@ -365,11 +365,10 @@
     return ::ConvertMaximum<hal_1_3::HalPolicy>(operation, model, data);
 }
 
-bool HalPolicy::ConvertReduce(const Operation& operation, const Model& model, ConversionData& data,
-                              ReduceOperation reduceOperation)
+bool HalPolicy::ConvertMean(const Operation& operation, const Model& model, ConversionData& data)
 {
-    ALOGV("hal_1_3::HalPolicy::ConvertReduce()");
-    return ::ConvertReduce<hal_1_3::HalPolicy>(operation, model, data, reduceOperation);
+    ALOGV("hal_1_3::HalPolicy::ConvertMean()");
+    return ::ConvertMean<hal_1_3::HalPolicy>(operation, model, data);
 }
 
 bool HalPolicy::ConvertMinimum(const Operation& operation, const Model& model, ConversionData& data)
@@ -426,6 +425,15 @@
     return ::ConvertRank<hal_1_3::HalPolicy>(operation, model, data);
 }
 
+bool HalPolicy::ConvertReduce(const Operation& operation,
+                              const Model& model,
+                              ConversionData& data,
+                              ReduceOperation reduceOperation)
+{
+    ALOGV("hal_1_3::HalPolicy::ConvertReduce()");
+    return ::ConvertReduce<hal_1_3::HalPolicy>(operation, model, data, reduceOperation);
+}
+
 bool HalPolicy::ConvertReLu(const Operation& operation, const Model& model, ConversionData& data)
 {
     ALOGV("hal_1_3::HalPolicy::ConvertReLu()");