diff options
| author | Matthew Bentham <matthew.bentham@arm.com> | 2019-07-30 08:24:12 +0000 |
|---|---|---|
| committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-07-30 15:58:30 +0000 |
| commit | 0cf01dce414db3ce9595a438cf3c3f10dd938450 (patch) | |
| tree | 900e14259365c2998a9ad75f3edd18b375c11fd6 /src/armnn/Layer.hpp | |
| parent | bc2e210785a63e8360839e4ded5d2c15c2dffaf5 (diff) | |
| download | armnn-0cf01dce414db3ce9595a438cf3c3f10dd938450.tar.gz | |
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 <Matthew.Bentham@arm.com>
Diffstat (limited to 'src/armnn/Layer.hpp')
| -rw-r--r-- | src/armnn/Layer.hpp | 1 |
1 files changed, 1 insertions, 0 deletions
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<InputSlot*>& GetConnections() const { return m_Connections; } + const std::vector<MemoryStrategy>& GetMemoryStrategies() const { return m_MemoryStrategies; } bool ValidateTensorShape(const TensorShape& shape) const; |