commit 51982472bfedf12e7d82cde6614617f94b2c86d0
author Ellen Norris-Thompson <ellen.norris-thompson@arm.com> Wed Jun 19 11:46:21 2019 +0100
committer Matteo Martincigh <matteo.martincigh@arm.com> Wed Jun 19 16:27:26 2019 +0000
tree 0e2bef4812e0a3128b47a81f393dd6583d45e4c7
parent a4812b6cd54e4dc4903f457066281d8bf0ccf448

IVGCVSW-3269 Add Serialization support for the new Prelu Activation layer

 * Adds serialization/deserialization support
 * Adds related unit test

Signed-off-by: Ellen Norris-Thompson <ellen.norris-thompson@arm.com>
Change-Id: I600322b03e51f443cbcd9262bb27e36e5fd95ae5

src/armnnSerializer/Serializer.cpp

diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp
index 012ed66..81231e4 100644
--- a/src/armnnSerializer/Serializer.cpp
+++ b/src/armnnSerializer/Serializer.cpp
@@ -712,7 +712,14 @@
 void SerializerVisitor::VisitPreluLayer(const armnn::IConnectableLayer* layer,
                                         const char* name)
 {
-    throw UnimplementedException("SerializerVisitor::VisitPreluLayer not yet implemented");
+    // Create FlatBuffer BaseLayer
+    auto flatBufferPreluBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Prelu);
+
+    // Create the FlatBuffer AdditionLayer
+    auto flatBufferPreluLayer = serializer::CreatePreluLayer(m_flatBufferBuilder, flatBufferPreluBaseLayer);
+
+    // Add the AnyLayer to the FlatBufferLayers
+    CreateAnyLayer(flatBufferPreluLayer.o, serializer::Layer::Layer_PreluLayer);
 }
 
 void SerializerVisitor::VisitQuantizeLayer(const armnn::IConnectableLayer *layer, const char *name)