aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/Serializer.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnnSerializer/Serializer.cpp')
-rw-r--r--src/armnnSerializer/Serializer.cpp6
1 files changed, 4 insertions, 2 deletions
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp
index dabe977179..efadbb38ca 100644
--- a/src/armnnSerializer/Serializer.cpp
+++ b/src/armnnSerializer/Serializer.cpp
@@ -380,9 +380,11 @@ void SerializerVisitor::VisitL2NormalizationLayer(const armnn::IConnectableLayer
// Create the FlatBuffer L2Normalization Descriptor
auto fbDescriptor = serializer::CreateL2NormalizationDescriptor(
- m_flatBufferBuilder, GetFlatBufferDataLayout(l2NormalizationDescriptor.m_DataLayout));
+ m_flatBufferBuilder,
+ GetFlatBufferDataLayout(l2NormalizationDescriptor.m_DataLayout),
+ l2NormalizationDescriptor.m_Eps);
- // Create Flatuffer layer
+ // Create FlatBuffer layer
auto fbLayer = serializer::CreateL2NormalizationLayer(m_flatBufferBuilder, fbBaseLayer, fbDescriptor);
CreateAnyLayer(fbLayer.o, serializer::Layer::Layer_L2NormalizationLayer);