diff options
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); |