diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-03-07 17:31:34 +0000 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-03-08 09:53:20 +0000 |
commit | 495701f016117b405d90a69de0814bf60751d78b (patch) | |
tree | 282f36b385e76ab8bc6add250250d10089e4c450 /src/armnnSerializer/Serializer.cpp | |
parent | ac25a1beda8da71a82c0cf2795e2a6eaaeaa26b1 (diff) | |
download | armnn-495701f016117b405d90a69de0814bf60751d78b.tar.gz |
IVGCVSW-2693 Serialize/de-serialize L2Normalization
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I6a53ac576260383f32fb0d878b42d1251ffde94a
Diffstat (limited to 'src/armnnSerializer/Serializer.cpp')
-rw-r--r-- | src/armnnSerializer/Serializer.cpp | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp index 3b71e5fc7d..eaf19d52b5 100644 --- a/src/armnnSerializer/Serializer.cpp +++ b/src/armnnSerializer/Serializer.cpp @@ -329,6 +329,23 @@ void SerializerVisitor::VisitGreaterLayer(const armnn::IConnectableLayer* layer, CreateAnyLayer(fbGreaterLayer.o, serializer::Layer::Layer_GreaterLayer); } +void SerializerVisitor::VisitL2NormalizationLayer(const armnn::IConnectableLayer* layer, + const armnn::L2NormalizationDescriptor& l2NormalizationDescriptor, + const char* name) +{ + // Create FlatBuffer BaseLayer + auto fbBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_L2Normalization); + + // Create the FlatBuffer L2Normalization Descriptor + auto fbDescriptor = serializer::CreateL2NormalizationDescriptor( + m_flatBufferBuilder, GetFlatBufferDataLayout(l2NormalizationDescriptor.m_DataLayout)); + + // Create Flatuffer layer + auto fbLayer = serializer::CreateL2NormalizationLayer(m_flatBufferBuilder, fbBaseLayer, fbDescriptor); + + CreateAnyLayer(fbLayer.o, serializer::Layer::Layer_L2NormalizationLayer); +} + void SerializerVisitor::VisitMaximumLayer(const armnn::IConnectableLayer* layer, const char* name) { auto fbMaximumBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Maximum); |