aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/Serializer.hpp
diff options
context:
space:
mode:
authorNina Drozd <nina.drozd@arm.com>2019-02-27 10:53:27 +0000
committerNina Drozd <nina.drozd@arm.com>2019-03-01 10:52:26 +0000
commit57728788f65656e4fa08923d12bee0de34a72fc7 (patch)
tree4d84f00762483cb26f6c2e880a07eb8b7699b7da /src/armnnSerializer/Serializer.hpp
parent377351e5420304668da92da4ee00a012923619d1 (diff)
downloadarmnn-57728788f65656e4fa08923d12bee0de34a72fc7.tar.gz
IVGCVSW-2700 Serialize/de-serialize the Normalization layer
Change-Id: Ib307ec6c28beb6c158d337678e67a2484c495a06 Signed-off-by: Nina Drozd <nina.drozd@arm.com>
Diffstat (limited to 'src/armnnSerializer/Serializer.hpp')
-rw-r--r--src/armnnSerializer/Serializer.hpp3
1 files changed, 3 insertions, 0 deletions
diff --git a/src/armnnSerializer/Serializer.hpp b/src/armnnSerializer/Serializer.hpp
index 3d6f1b5700..7e6097c465 100644
--- a/src/armnnSerializer/Serializer.hpp
+++ b/src/armnnSerializer/Serializer.hpp
@@ -118,6 +118,9 @@ public:
const armnn::SpaceToBatchNdDescriptor& spaceToBatchNdDescriptor,
const char* name = nullptr) override;
+ void VisitNormalizationLayer(const armnn::IConnectableLayer* layer,
+ const armnn::NormalizationDescriptor& normalizationDescriptor,
+ const char* name = nullptr) override;
private:
/// Creates the Input Slots and Output Slots and LayerBase for the layer.