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/armnnDeserializer/Deserializer.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/armnnDeserializer/Deserializer.cpp')
-rw-r--r-- | src/armnnDeserializer/Deserializer.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/armnnDeserializer/Deserializer.cpp b/src/armnnDeserializer/Deserializer.cpp index b23ed97b4c..90ca6d3f9a 100644 --- a/src/armnnDeserializer/Deserializer.cpp +++ b/src/armnnDeserializer/Deserializer.cpp @@ -1187,6 +1187,7 @@ void Deserializer::ParseL2Normalization(GraphPtr graph, unsigned int layerIndex) auto layerName = GetLayerName(graph, layerIndex); armnn::L2NormalizationDescriptor descriptor; descriptor.m_DataLayout = ToDataLayout(flatBufferDescriptor->dataLayout()); + descriptor.m_Eps = flatBufferDescriptor->eps(); IConnectableLayer* layer = m_Network->AddL2NormalizationLayer(descriptor, layerName.c_str()); layer->GetOutputSlot(0).SetTensorInfo(outputInfo); |