From 09fa24d2f4b0177d55800bd01ec52c337701ef0a Mon Sep 17 00:00:00 2001 From: Nabeel Ahmad Date: Thu, 16 Jun 2022 13:55:00 +0100 Subject: NNXSW-3858: Get non-const IConnectableLayer from I/O slots * Added non-const variants of existing const member functions in IInputSlot and IOutputSlot to retrieve non-const IConnectableLayer Signed-off-by: Nabeel Ahmad Change-Id: Ic3388b578324edb4d2cca36acce6560ad1ce83c5 --- src/armnn/Layer.hpp | 2 ++ 1 file changed, 2 insertions(+) (limited to 'src/armnn/Layer.hpp') diff --git a/src/armnn/Layer.hpp b/src/armnn/Layer.hpp index 767cf97eb3..5edf66cabb 100644 --- a/src/armnn/Layer.hpp +++ b/src/armnn/Layer.hpp @@ -57,6 +57,7 @@ public: OutputSlot* GetConnectedOutputSlot() { return m_Connection; } const IConnectableLayer& GetOwningIConnectableLayer() const override; + IConnectableLayer& GetOwningIConnectableLayer() override; /// Links the slot to an output slot or breaks an existing link if passing nullptr. void SetConnection(OutputSlot* source) @@ -118,6 +119,7 @@ public: Layer& GetOwningLayer() const { return m_OwningLayer; } const IConnectableLayer& GetOwningIConnectableLayer() const override; + IConnectableLayer& GetOwningIConnectableLayer() override; LayerGuid GetOwningLayerGuid() const override; -- cgit v1.2.1