diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-10-17 16:11:54 +0100 |
---|---|---|
committer | Áron Virginás-Tar <aron.virginas-tar@arm.com> | 2019-10-21 10:54:45 +0000 |
commit | e80ebd101b516751a798aa1b1d669e9117a32266 (patch) | |
tree | 9f113060aaa2157941f726af6ec93ea8e9357df8 /src/armnnSerializer/SerializerUtils.cpp | |
parent | 77bfb5e32faadb1383d48364a6f54adbff84ad80 (diff) | |
download | armnn-e80ebd101b516751a798aa1b1d669e9117a32266.tar.gz |
IVGCVSW-3992 Add serialization support for ComparisonLayer
* Added serialization support and serialization-deserialization test
for Comparison
* Added backward compatibility tests for Equal and Greater, to make
sure they are serialized and deserialized as Comparison
* Refactored serialization tests and reduced code duplication by
taking advantage of operator==() recently added for Descriptors
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: Id0fd94ef4b17e4e51c8005e585ea3c47f9c1bd8d
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 |