diff options
author | Saoirse Stewart <saoirse.stewart@arm.com> | 2019-02-18 15:24:53 +0000 |
---|---|---|
committer | Aron Virginas-Tar <aron.virginas-tar@arm.com> | 2019-02-19 11:52:27 +0000 |
commit | 3166c3edeb64d834ba27031ddd39b5b1f940b604 (patch) | |
tree | 2789010d0878d64442f51ba0edbd8f159d1a32a0 /src/armnnDeserializeParser/DeserializeParser.hpp | |
parent | a6b504a8925174739f5a064cf77d1563cca38708 (diff) | |
download | armnn-3166c3edeb64d834ba27031ddd39b5b1f940b604.tar.gz |
IVGCVSW-2645 Add Serializer & Deserializer for Pooling2d
Change-Id: Iba41da3cccd539a0175f2ed0ff9a8b6a23c5fb6f
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Signed-off-by: Saoirse Stewart <saoirse.stewart@arm.com>
Diffstat (limited to 'src/armnnDeserializeParser/DeserializeParser.hpp')
-rw-r--r-- | src/armnnDeserializeParser/DeserializeParser.hpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/armnnDeserializeParser/DeserializeParser.hpp b/src/armnnDeserializeParser/DeserializeParser.hpp index ddd02abede..1edb5a9f23 100644 --- a/src/armnnDeserializeParser/DeserializeParser.hpp +++ b/src/armnnDeserializeParser/DeserializeParser.hpp @@ -17,6 +17,7 @@ public: // Shorthands for deserializer types using GraphPtr = const armnn::armnnSerializer::SerializedGraph *; using TensorRawPtr = const armnn::armnnSerializer::TensorInfo *; + using PoolingDescriptor = const armnn::armnnSerializer::Pooling2dDescriptor *; using TensorRawPtrVector = std::vector<TensorRawPtr>; using LayerRawPtr = const armnn::armnnSerializer::LayerBase *; using LayerBaseRawPtr = const armnn::armnnSerializer::LayerBase *; @@ -50,6 +51,8 @@ public: static LayerBaseRawPtrVector GetGraphOutputs(const GraphPtr& graphPtr); static LayerBaseRawPtr GetBaseLayer(const GraphPtr& graphPtr, unsigned int layerIndex); static int32_t GetBindingLayerInfo(const GraphPtr& graphPtr, unsigned int layerIndex); + armnn::Pooling2dDescriptor GetPoolingDescriptor(PoolingDescriptor pooling2dDescriptor, + unsigned int layerIndex); private: // No copying allowed until it is wanted and properly implemented @@ -65,6 +68,7 @@ private: void ParseUnsupportedLayer(unsigned int layerIndex); void ParseAdd(unsigned int layerIndex); void ParseMultiplication(unsigned int layerIndex); + void ParsePooling2d(unsigned int layerIndex); void ParseSoftmax(unsigned int layerIndex); void RegisterOutputSlotOfConnection(uint32_t connectionIndex, armnn::IOutputSlot* slot); @@ -82,6 +86,7 @@ private: armnn::INetworkPtr m_Network; GraphPtr m_Graph; std::vector<LayerParsingFunction> m_ParserFunctions; + std::string m_layerName; /// This holds the data of the file that was read in from CreateNetworkFromBinaryFile /// Needed for m_Graph to point to |