diff options
Diffstat (limited to 'src/armnnSerializer/SerializerUtils.cpp')
| -rw-r--r-- | src/armnnSerializer/SerializerUtils.cpp | 22 |
1 files changed, 20 insertions, 2 deletions
diff --git a/src/armnnSerializer/SerializerUtils.cpp b/src/armnnSerializer/SerializerUtils.cpp index 9790d6e23b..719f8a92dd 100644 --- a/src/armnnSerializer/SerializerUtils.cpp +++ b/src/armnnSerializer/SerializerUtils.cpp @@ -8,7 +8,25 @@ namespace armnnSerializer { -using namespace armnn; +armnnSerializer::ComparisonOperation GetFlatBufferComparisonOperation(armnn::ComparisonOperation comparisonOperation) +{ + switch (comparisonOperation) + { + case armnn::ComparisonOperation::Equal: + return armnnSerializer::ComparisonOperation::ComparisonOperation_Equal; + case armnn::ComparisonOperation::Greater: + return armnnSerializer::ComparisonOperation::ComparisonOperation_Greater; + case armnn::ComparisonOperation::GreaterOrEqual: + return armnnSerializer::ComparisonOperation::ComparisonOperation_GreaterOrEqual; + case armnn::ComparisonOperation::Less: + return armnnSerializer::ComparisonOperation::ComparisonOperation_Less; + case armnn::ComparisonOperation::LessOrEqual: + return armnnSerializer::ComparisonOperation::ComparisonOperation_LessOrEqual; + case armnn::ComparisonOperation::NotEqual: + default: + return armnnSerializer::ComparisonOperation::ComparisonOperation_NotEqual; + } +} armnnSerializer::ConstTensorData GetFlatBufferConstTensorData(armnn::DataType dataType) { @@ -137,4 +155,4 @@ armnnSerializer::ResizeMethod GetFlatBufferResizeMethod(armnn::ResizeMethod meth } } -} // namespace armnnSerializer
\ No newline at end of file +} // namespace armnnSerializer |