diff options
Diffstat (limited to 'src/armnnSerializer/SerializerUtils.cpp')
-rw-r--r-- | src/armnnSerializer/SerializerUtils.cpp | 17 |
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 |