aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/SerializerUtils.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnnSerializer/SerializerUtils.cpp')
-rw-r--r--src/armnnSerializer/SerializerUtils.cpp17
1 files changed, 17 insertions, 0 deletions
diff --git a/src/armnnSerializer/SerializerUtils.cpp b/src/armnnSerializer/SerializerUtils.cpp
index 045d6aac5c..32ac75e024 100644
--- a/src/armnnSerializer/SerializerUtils.cpp
+++ b/src/armnnSerializer/SerializerUtils.cpp
@@ -198,4 +198,21 @@ armnnSerializer::ResizeMethod GetFlatBufferResizeMethod(armnn::ResizeMethod meth
}
}
+armnnSerializer::ReduceOperation GetFlatBufferReduceOperation(armnn::ReduceOperation reduceOperation)
+{
+ switch (reduceOperation)
+ {
+ case armnn::ReduceOperation::Sum:
+ return armnnSerializer::ReduceOperation::ReduceOperation_Sum;
+ case armnn::ReduceOperation::Max:
+ return armnnSerializer::ReduceOperation::ReduceOperation_Max;
+ case armnn::ReduceOperation::Mean:
+ return armnnSerializer::ReduceOperation::ReduceOperation_Mean;
+ case armnn::ReduceOperation::Min:
+ return armnnSerializer::ReduceOperation::ReduceOperation_Min;
+ default:
+ return armnnSerializer::ReduceOperation::ReduceOperation_Sum;
+ }
+}
+
} // namespace armnnSerializer