From b619b42118f54c2db1946aef46477926efa4b75c Mon Sep 17 00:00:00 2001 From: Francis Murtagh Date: Mon, 27 Jun 2022 12:44:50 +0100 Subject: 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 --- src/armnn/Layer.hpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/armnn/Layer.hpp') 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; -- cgit v1.2.1