diff options
author | Éanna Ó Catháin <eanna.ocathain@arm.com> | 2019-02-25 16:26:29 +0000 |
---|---|---|
committer | Eanna O Cathain Arm <eanna.ocathain@arm.com> | 2019-02-25 17:28:08 +0000 |
commit | 633f859043522adb7629f3f7ea617cd65b3713f2 (patch) | |
tree | 39fdc2fdc5709076c1852f85ee8aa6702c83dae6 /src/armnnDeserializer/Deserializer.hpp | |
parent | 268509ac1750c6c8d7c6f0debd9dca7e42612341 (diff) | |
download | armnn-633f859043522adb7629f3f7ea617cd65b3713f2.tar.gz |
IVGCVSW-2751 When layers are Deserialized the layer name is lost
* Fixed defect
* Added unit tests to check names
Change-Id: I67211110b90aabff6e00bccdadd1145423b07523
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Signed-off-by: Éanna Ó Catháin <eanna.ocathain@arm.com>
Diffstat (limited to 'src/armnnDeserializer/Deserializer.hpp')
-rw-r--r-- | src/armnnDeserializer/Deserializer.hpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/armnnDeserializer/Deserializer.hpp b/src/armnnDeserializer/Deserializer.hpp index 1e775d0c4a..94318e4062 100644 --- a/src/armnnDeserializer/Deserializer.hpp +++ b/src/armnnDeserializer/Deserializer.hpp @@ -50,6 +50,7 @@ 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); + static std::string GetLayerName(const GraphPtr& graph, unsigned int index); armnn::Pooling2dDescriptor GetPoolingDescriptor(PoolingDescriptor pooling2dDescriptor, unsigned int layerIndex); static armnn::TensorInfo OutputShapeOfReshape(const armnn::TensorInfo & inputTensorInfo, @@ -92,7 +93,6 @@ private: /// The network we're building. Gets cleared after it is passed to the user armnn::INetworkPtr m_Network; std::vector<LayerParsingFunction> m_ParserFunctions; - std::string m_layerName; using NameToBindingInfo = std::pair<std::string, BindingPointInfo >; std::vector<NameToBindingInfo> m_InputBindings; |