commit 57728788f65656e4fa08923d12bee0de34a72fc7
author Nina Drozd <nina.drozd@arm.com> Wed Feb 27 10:53:27 2019 +0000
committer Nina Drozd <nina.drozd@arm.com> Fri Mar 01 10:52:26 2019 +0000
tree 4d84f00762483cb26f6c2e880a07eb8b7699b7da
parent 377351e5420304668da92da4ee00a012923619d1

IVGCVSW-2700 Serialize/de-serialize the Normalization layer

Change-Id: Ib307ec6c28beb6c158d337678e67a2484c495a06
Signed-off-by: Nina Drozd <nina.drozd@arm.com>

src/armnnSerializer/Serializer.cpp

diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp
index a94a319..2000726 100644
--- a/src/armnnSerializer/Serializer.cpp
+++ b/src/armnnSerializer/Serializer.cpp
@@ -470,6 +470,29 @@
     CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_SpaceToBatchNdLayer);
 }
 
+void SerializerVisitor::VisitNormalizationLayer(const armnn::IConnectableLayer* layer,
+                                                const armnn::NormalizationDescriptor& descriptor,
+                                                const char* name)
+{
+    auto fbNormalizationBaseLayer  = CreateLayerBase(layer, serializer::LayerType::LayerType_Normalization);
+
+    auto fbNormalizationDescriptor = serializer::CreateNormalizationDescriptor(
+        m_flatBufferBuilder,
+        GetFlatBufferNormalizationAlgorithmChannel(descriptor.m_NormChannelType),
+        GetFlatBufferNormalizationAlgorithmMethod(descriptor.m_NormMethodType),
+        descriptor.m_NormSize,
+        descriptor.m_Alpha,
+        descriptor.m_Beta,
+        descriptor.m_K,
+        GetFlatBufferDataLayout(descriptor.m_DataLayout));
+
+    auto flatBufferLayer = serializer::CreateNormalizationLayer(m_flatBufferBuilder,
+                                                                fbNormalizationBaseLayer,
+                                                                fbNormalizationDescriptor);
+
+    CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_NormalizationLayer);
+}
+
 fb::Offset<serializer::LayerBase> SerializerVisitor::CreateLayerBase(const IConnectableLayer* layer,
                                                                      const serializer::LayerType layerType)
 {