From 4e3e831da1d6d85dffffacf57e9de8fc891b7e58 Mon Sep 17 00:00:00 2001
From: Teresa Charlin <teresa.charlinreyes@arm.com>
Date: Thu, 5 Aug 2021 12:34:37 +0100
Subject: IVGCVSW-6262 Add support for Reduce Prod

* Tflite parser
* Tflite delegate
* Serializer
* Deserializer
* Ref, CpuAcc and GpuAcc workloads

Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I601a9ee1680b372c7955d9a628857d08c3cfd377
---
 src/armnnSerializer/ArmnnSchema.fbs         |  3 ++-
 src/armnnSerializer/ArmnnSchema_generated.h | 11 +++++++----
 src/armnnSerializer/SerializerUtils.cpp     |  2 ++
 3 files changed, 11 insertions(+), 5 deletions(-)

(limited to 'src/armnnSerializer')

diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index 85435a366f..a285a110cc 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -52,7 +52,8 @@ enum ReduceOperation: byte {
     Sum  = 0,
     Max  = 1,
     Mean = 2,
-    Min  = 3
+    Min  = 3,
+    Prod  = 4
 }
 
 enum ResizeMethod: byte {
diff --git a/src/armnnSerializer/ArmnnSchema_generated.h b/src/armnnSerializer/ArmnnSchema_generated.h
index ca2bf0c003..cf28a7a313 100644
--- a/src/armnnSerializer/ArmnnSchema_generated.h
+++ b/src/armnnSerializer/ArmnnSchema_generated.h
@@ -559,26 +559,29 @@ enum ReduceOperation {
   ReduceOperation_Max = 1,
   ReduceOperation_Mean = 2,
   ReduceOperation_Min = 3,
+  ReduceOperation_Prod = 4,
   ReduceOperation_MIN = ReduceOperation_Sum,
-  ReduceOperation_MAX = ReduceOperation_Min
+  ReduceOperation_MAX = ReduceOperation_Prod
 };
 
-inline const ReduceOperation (&EnumValuesReduceOperation())[4] {
+inline const ReduceOperation (&EnumValuesReduceOperation())[5] {
   static const ReduceOperation values[] = {
     ReduceOperation_Sum,
     ReduceOperation_Max,
     ReduceOperation_Mean,
-    ReduceOperation_Min
+    ReduceOperation_Min,
+    ReduceOperation_Prod
   };
   return values;
 }
 
 inline const char * const *EnumNamesReduceOperation() {
-  static const char * const names[5] = {
+  static const char * const names[6] = {
     "Sum",
     "Max",
     "Mean",
     "Min",
+    "Prod",
     nullptr
   };
   return names;
diff --git a/src/armnnSerializer/SerializerUtils.cpp b/src/armnnSerializer/SerializerUtils.cpp
index 1df8d4e608..85ce01d132 100644
--- a/src/armnnSerializer/SerializerUtils.cpp
+++ b/src/armnnSerializer/SerializerUtils.cpp
@@ -219,6 +219,8 @@ armnnSerializer::ReduceOperation GetFlatBufferReduceOperation(armnn::ReduceOpera
             return armnnSerializer::ReduceOperation::ReduceOperation_Mean;
         case armnn::ReduceOperation::Min:
             return armnnSerializer::ReduceOperation::ReduceOperation_Min;
+        case armnn::ReduceOperation::Prod:
+            return armnnSerializer::ReduceOperation::ReduceOperation_Prod;
         default:
             return armnnSerializer::ReduceOperation::ReduceOperation_Sum;
     }
-- 
cgit v1.2.1