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/Serializer.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/Serializer.cpp')
-rw-r--r-- | src/armnnSerializer/Serializer.cpp | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp index 5c9855f87e..6122351812 100644 --- a/src/armnnSerializer/Serializer.cpp +++ b/src/armnnSerializer/Serializer.cpp @@ -244,7 +244,13 @@ void SerializerVisitor::VisitComparisonLayer(const armnn::IConnectableLayer* lay const armnn::ComparisonDescriptor& descriptor, const char* name) { - throw armnn::UnimplementedException("SerializerVisitor::VisitComparisonLayer() is not implemented"); + auto fbBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Comparison); + auto fbDescriptor = serializer::CreateComparisonDescriptor( + m_flatBufferBuilder, + GetFlatBufferComparisonOperation(descriptor.m_Operation)); + + auto fbLayer = serializer::CreateComparisonLayer(m_flatBufferBuilder, fbBaseLayer, fbDescriptor); + CreateAnyLayer(fbLayer.o, serializer::Layer::Layer_ComparisonLayer); } // Build FlatBuffer for Constant Layer |