diff options
Diffstat (limited to 'src/armnnSerializer')
-rw-r--r-- | src/armnnSerializer/ArmnnSchema.fbs | 3 | ||||
-rw-r--r-- | src/armnnSerializer/ArmnnSchema_generated.h | 11 | ||||
-rw-r--r-- | src/armnnSerializer/SerializerUtils.cpp | 2 |
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; } |