diff options
Diffstat (limited to 'src/armnnSerializer/Serializer.cpp')
-rw-r--r-- | src/armnnSerializer/Serializer.cpp | 6 |
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); |