IVGCVSW-3934 Add serialization support for INSTANCE_NORMALIZATION
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: If9b4d30cbd625206ec1c7d37dd8b449983442147
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp
index 84a1b6b..5949d1d 100644
--- a/src/armnnSerializer/Serializer.cpp
+++ b/src/armnnSerializer/Serializer.cpp
@@ -431,7 +431,17 @@
const armnn::InstanceNormalizationDescriptor& instanceNormalizationDescriptor,
const char* name)
{
- throw UnimplementedException("SerializerVisitor::InstanceNormalizationLayer is not implemented");
+ auto fbDescriptor = serializer::CreateInstanceNormalizationDescriptor(
+ m_flatBufferBuilder,
+ instanceNormalizationDescriptor.m_Gamma,
+ instanceNormalizationDescriptor.m_Beta,
+ instanceNormalizationDescriptor.m_Eps,
+ GetFlatBufferDataLayout(instanceNormalizationDescriptor.m_DataLayout));
+
+ auto fbBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_InstanceNormalization);
+ auto fbLayer = serializer::CreateInstanceNormalizationLayer(m_flatBufferBuilder, fbBaseLayer, fbDescriptor);
+
+ CreateAnyLayer(fbLayer.o, serializer::Layer::Layer_InstanceNormalizationLayer);
}
void SerializerVisitor::VisitL2NormalizationLayer(const armnn::IConnectableLayer* layer,