diff options
author | Ferran Balaguer <ferran.balaguer@arm.com> | 2019-06-18 16:25:06 +0100 |
---|---|---|
committer | Ferran Balaguer Arm <ferran.balaguer@arm.com> | 2019-06-21 13:38:35 +0000 |
commit | 0dcffec80292cd2e0e7c2736fd3db63abd7c3f64 (patch) | |
tree | 1796f284f8537b8235082f9684f68404ccd1e926 /src/armnnSerializer/Serializer.cpp | |
parent | 639fb0437d1a5a8a6ea737fed5a16b554dfffead (diff) | |
download | armnn-0dcffec80292cd2e0e7c2736fd3db63abd7c3f64.tar.gz |
IVGCVSW-3291 Add L2Normalization epsilon value to serialization
Signed-off-by: Ferran Balaguer <ferran.balaguer@arm.com>
Change-Id: Icfff3fb596a03c126a42b1d0c254a68e498df734
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); |