From 633f859043522adb7629f3f7ea617cd65b3713f2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C3=89anna=20=C3=93=20Cath=C3=A1in?= Date: Mon, 25 Feb 2019 16:26:29 +0000 Subject: IVGCVSW-2751 When layers are Deserialized the layer name is lost MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * Fixed defect * Added unit tests to check names Change-Id: I67211110b90aabff6e00bccdadd1145423b07523 Signed-off-by: Mike Kelly Signed-off-by: Éanna Ó Catháin --- src/armnnDeserializer/Deserializer.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src/armnnDeserializer/Deserializer.hpp') 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 m_ParserFunctions; - std::string m_layerName; using NameToBindingInfo = std::pair; std::vector m_InputBindings; -- cgit v1.2.1