From 0cf01dce414db3ce9595a438cf3c3f10dd938450 Mon Sep 17 00:00:00 2001 From: Matthew Bentham Date: Tue, 30 Jul 2019 08:24:12 +0000 Subject: IVGCVSW-3581 Fix AddCopyLayers and associated tests Take a copy of the MemoryStrategies for a layer before inserting new connections. Use the copy when looking up the original MemoryStrategies during the graph transformation. Fix the unit tests for AddCopyLayers to have cases where copies are needed. Fix the validation for clarity and correctness - was previously comparing Layers by pointer when it should have been by name (as it was comparing with a cloned graph). Change-Id: Ie282dc11913e977b8151ce1ad8bfba5e11617d40 Signed-off-by: Matthew Bentham --- src/armnn/Layer.hpp | 1 + 1 file changed, 1 insertion(+) (limited to 'src/armnn/Layer.hpp') diff --git a/src/armnn/Layer.hpp b/src/armnn/Layer.hpp index 1ddbc00bc7..b90d040475 100644 --- a/src/armnn/Layer.hpp +++ b/src/armnn/Layer.hpp @@ -123,6 +123,7 @@ public: void Disconnect(InputSlot& slot); const std::vector& GetConnections() const { return m_Connections; } + const std::vector& GetMemoryStrategies() const { return m_MemoryStrategies; } bool ValidateTensorShape(const TensorShape& shape) const; -- cgit v1.2.1