diff options
| author | Saoirse Stewart <saoirse.stewart@arm.com> | 2019-02-19 15:54:14 +0000 |
|---|---|---|
| committer | Saoirse Stewart Arm <saoirse.stewart@arm.com> | 2019-02-19 15:55:46 +0000 |
| commit | 263829c2163d79a28f98f24f9dd1e52e1c3cbbef (patch) | |
| tree | bd904ce4b8aeaa14bc0622bbacefda26011733f2 /src/armnnDeserializeParser/DeserializeParser.hpp | |
| parent | 4fbae33571871ce584e421657e8ffba299e89d67 (diff) | |
| download | armnn-263829c2163d79a28f98f24f9dd1e52e1c3cbbef.tar.gz | |
IVGCVSW-2642 Add Reshape to Serializer and Deserializer
Change-Id: Iccded3c6e3c0713c44f43231981440420591f94e
Signed-off-by: Saoirse Stewart <saoirse.stewart@arm.com>
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Diffstat (limited to 'src/armnnDeserializeParser/DeserializeParser.hpp')
| -rw-r--r-- | src/armnnDeserializeParser/DeserializeParser.hpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/armnnDeserializeParser/DeserializeParser.hpp b/src/armnnDeserializeParser/DeserializeParser.hpp index 1edb5a9f23..666cbca33c 100644 --- a/src/armnnDeserializeParser/DeserializeParser.hpp +++ b/src/armnnDeserializeParser/DeserializeParser.hpp @@ -53,6 +53,8 @@ public: static int32_t GetBindingLayerInfo(const GraphPtr& graphPtr, unsigned int layerIndex); armnn::Pooling2dDescriptor GetPoolingDescriptor(PoolingDescriptor pooling2dDescriptor, unsigned int layerIndex); + static armnn::TensorInfo OutputShapeOfReshape(const armnn::TensorInfo & inputTensorInfo, + const std::vector<uint32_t> & targetDimsIn); private: // No copying allowed until it is wanted and properly implemented @@ -69,6 +71,7 @@ private: void ParseAdd(unsigned int layerIndex); void ParseMultiplication(unsigned int layerIndex); void ParsePooling2d(unsigned int layerIndex); + void ParseReshape(unsigned int layerIndex); void ParseSoftmax(unsigned int layerIndex); void RegisterOutputSlotOfConnection(uint32_t connectionIndex, armnn::IOutputSlot* slot); |