diff options
author | Francis Murtagh <francis.murtagh@arm.com> | 2022-06-27 12:44:50 +0100 |
---|---|---|
committer | Nikhil Raj <nikhil.raj@arm.com> | 2022-07-08 15:21:39 +0100 |
commit | b619b42118f54c2db1946aef46477926efa4b75c (patch) | |
tree | 00a101ce4a4db89481efe36ef312e6a1aae78126 /src/armnn/Layer.hpp | |
parent | f4ccb1f6339a1e9ed573f188e7f14353167b5749 (diff) | |
download | armnn-b619b42118f54c2db1946aef46477926efa4b75c.tar.gz |
IVGCVSW-7034 Modified SubgraphView returned by GetWorkingCopy()
* Add virtual GetSlotIndex to IInputSlot
* Fix logic in GetWorkingCopy to use index of slots; so as not
to add slots to cloned subgraphView if not in original subgraphView
* Add test to cover cases when not all inputSlots to subgraphView layer
are part of the original subgraphView
* Mark SubgraphView::GetWorkingCopy() as const
Change-Id: I1d540f84c57f97f6c834ec06ca13393ffa55d379
Diffstat (limited to 'src/armnn/Layer.hpp')
-rw-r--r-- | src/armnn/Layer.hpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/armnn/Layer.hpp b/src/armnn/Layer.hpp index 12c782c965..b144c78889 100644 --- a/src/armnn/Layer.hpp +++ b/src/armnn/Layer.hpp @@ -51,7 +51,7 @@ public: ~InputSlot(); Layer& GetOwningLayer() const { return m_OwningLayer; } - unsigned int GetSlotIndex() const { return m_SlotIndex; } + unsigned int GetSlotIndex() const override { return m_SlotIndex; } const OutputSlot* GetConnectedOutputSlot() const { return m_Connection; } OutputSlot* GetConnectedOutputSlot() { return m_Connection; } @@ -73,7 +73,7 @@ public: // Inserts single-output existing layer at this point in the graph. void Insert(Layer& layer); - // IInputSlot + // InputSlot const IOutputSlot* GetConnection() const override; IOutputSlot* GetConnection() override; |