From 51982472bfedf12e7d82cde6614617f94b2c86d0 Mon Sep 17 00:00:00 2001 From: Ellen Norris-Thompson Date: Wed, 19 Jun 2019 11:46:21 +0100 Subject: 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 Change-Id: I600322b03e51f443cbcd9262bb27e36e5fd95ae5 --- src/armnnSerializer/Serializer.cpp | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) (limited to 'src/armnnSerializer/Serializer.cpp') diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp index 012ed666f1..81231e4eba 100644 --- a/src/armnnSerializer/Serializer.cpp +++ b/src/armnnSerializer/Serializer.cpp @@ -712,7 +712,14 @@ void SerializerVisitor::VisitPooling2dLayer(const armnn::IConnectableLayer* laye 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) -- cgit v1.2.1