diff options
author | Nina Drozd <nina.drozd@arm.com> | 2019-02-27 10:53:27 +0000 |
---|---|---|
committer | Nina Drozd <nina.drozd@arm.com> | 2019-03-01 10:52:26 +0000 |
commit | 57728788f65656e4fa08923d12bee0de34a72fc7 (patch) | |
tree | 4d84f00762483cb26f6c2e880a07eb8b7699b7da /src/armnnDeserializer/Deserializer.hpp | |
parent | 377351e5420304668da92da4ee00a012923619d1 (diff) | |
download | armnn-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/armnnDeserializer/Deserializer.hpp')
-rw-r--r-- | src/armnnDeserializer/Deserializer.hpp | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/src/armnnDeserializer/Deserializer.hpp b/src/armnnDeserializer/Deserializer.hpp index 1dd7ec5284..fba8b88044 100644 --- a/src/armnnDeserializer/Deserializer.hpp +++ b/src/armnnDeserializer/Deserializer.hpp @@ -19,6 +19,7 @@ public: using GraphPtr = const armnnSerializer::SerializedGraph *; using TensorRawPtr = const armnnSerializer::TensorInfo *; using PoolingDescriptor = const armnnSerializer::Pooling2dDescriptor *; + using NormalizationDescriptorPtr = const armnnSerializer::NormalizationDescriptor *; using TensorRawPtrVector = std::vector<TensorRawPtr>; using LayerRawPtr = const armnnSerializer::LayerBase *; using LayerBaseRawPtr = const armnnSerializer::LayerBase *; @@ -51,8 +52,10 @@ public: static LayerBaseRawPtr GetBaseLayer(const GraphPtr& graphPtr, unsigned int layerIndex); static int32_t GetBindingLayerInfo(const GraphPtr& graphPtr, unsigned int layerIndex); static std::string GetLayerName(const GraphPtr& graph, unsigned int index); - armnn::Pooling2dDescriptor GetPoolingDescriptor(PoolingDescriptor pooling2dDescriptor, - unsigned int layerIndex); + static armnn::Pooling2dDescriptor GetPoolingDescriptor(PoolingDescriptor pooling2dDescriptor, + unsigned int layerIndex); + static armnn::NormalizationDescriptor GetNormalizationDescriptor( + NormalizationDescriptorPtr normalizationDescriptor, unsigned int layerIndex); static armnn::TensorInfo OutputShapeOfReshape(const armnn::TensorInfo & inputTensorInfo, const std::vector<uint32_t> & targetDimsIn); @@ -80,6 +83,7 @@ private: void ParseMinimum(GraphPtr graph, unsigned int layerIndex); void ParseMaximum(GraphPtr graph, unsigned int layerIndex); void ParseMultiplication(GraphPtr graph, unsigned int layerIndex); + void ParseNormalization(GraphPtr graph, unsigned int layerIndex); void ParsePermute(GraphPtr graph, unsigned int layerIndex); void ParsePooling2d(GraphPtr graph, unsigned int layerIndex); void ParseReshape(GraphPtr graph, unsigned int layerIndex); |