From fc413c0c977e6c9680a2aa6546e977be0a2efdb9 Mon Sep 17 00:00:00 2001 From: Aron Virginas-Tar Date: Wed, 13 Feb 2019 15:41:52 +0000 Subject: IVGCVSW-2644 Add Serializer & Deserializer for Softmax Change-Id: Ifea2108e173d2b602162fe53b880a68e1c715510 Signed-off-by: Aron Virginas-Tar --- src/armnnSerializer/Serializer.cpp | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) (limited to 'src/armnnSerializer/Serializer.cpp') diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp index 79ad7e19af..ba4b36934c 100644 --- a/src/armnnSerializer/Serializer.cpp +++ b/src/armnnSerializer/Serializer.cpp @@ -119,6 +119,27 @@ void SerializerVisitor::VisitMultiplicationLayer(const IConnectableLayer* layer, CreateAnyLayer(flatBufferMultiplicationLayer.o, serializer::Layer::Layer_MultiplicationLayer); } +// Build FlatBuffer for Softmax Layer +void SerializerVisitor::VisitSoftmaxLayer(const IConnectableLayer* layer, + const SoftmaxDescriptor& softmaxDescriptor, + const char* name) +{ + // Create FlatBuffer BaseLayer + auto flatBufferSoftmaxBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Softmax); + + // Create the FlatBuffer SoftmaxDescriptor + auto flatBufferSoftmaxDesc = + serializer::CreateSoftmaxDescriptor(m_flatBufferBuilder, softmaxDescriptor.m_Beta); + + // Create the FlatBuffer SoftmaxLayer + auto flatBufferSoftmaxLayer = + serializer::CreateSoftmaxLayer(m_flatBufferBuilder, + flatBufferSoftmaxBaseLayer, + flatBufferSoftmaxDesc); + + CreateAnyLayer(flatBufferSoftmaxLayer.o, serializer::Layer::Layer_SoftmaxLayer); +} + fb::Offset SerializerVisitor::CreateLayerBase(const IConnectableLayer* layer, const serializer::LayerType layerType) { -- cgit v1.2.1