aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer
diff options
context:
space:
mode:
authorTeresa Charlin <teresa.charlinreyes@arm.com>2021-08-05 12:34:37 +0100
committerTeresaARM <teresa.charlinreyes@arm.com>2021-09-03 08:41:21 +0000
commit4e3e831da1d6d85dffffacf57e9de8fc891b7e58 (patch)
tree9a3653729feba788dcfbbdc5255ad379cbbf597d /src/armnnSerializer
parent14bef9f83f7cd58e5038ae7432d75da2d50e7b68 (diff)
downloadarmnn-4e3e831da1d6d85dffffacf57e9de8fc891b7e58.tar.gz
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
Diffstat (limited to 'src/armnnSerializer')
-rw-r--r--src/armnnSerializer/ArmnnSchema.fbs3
-rw-r--r--src/armnnSerializer/ArmnnSchema_generated.h11
-rw-r--r--src/armnnSerializer/SerializerUtils.cpp2
3 files changed, 11 insertions, 5 deletions
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;
}