commit e80ebd101b516751a798aa1b1d669e9117a32266
author Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> Thu Oct 17 16:11:54 2019 +0100
committer Áron Virginás-Tar <aron.virginas-tar@arm.com> Mon Oct 21 10:54:45 2019 +0000
tree 9f113060aaa2157941f726af6ec93ea8e9357df8
parent 77bfb5e32faadb1383d48364a6f54adbff84ad80

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

src/armnnSerializer/Serializer.cpp

diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp
index 5c9855f..6122351 100644
--- a/src/armnnSerializer/Serializer.cpp
+++ b/src/armnnSerializer/Serializer.cpp
@@ -244,7 +244,13 @@
                                              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