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.cpp | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'src/armnn/Layer.cpp') diff --git a/src/armnn/Layer.cpp b/src/armnn/Layer.cpp index b1d495244d..19337dca5c 100644 --- a/src/armnn/Layer.cpp +++ b/src/armnn/Layer.cpp @@ -543,9 +543,19 @@ const IConnectableLayer& OutputSlot::GetOwningIConnectableLayer() const return m_OwningLayer; } +IConnectableLayer& OutputSlot::GetOwningIConnectableLayer() +{ + return m_OwningLayer; +} + const IConnectableLayer& InputSlot::GetOwningIConnectableLayer() const { return m_OwningLayer; } +IConnectableLayer& InputSlot::GetOwningIConnectableLayer() +{ + return m_OwningLayer; +} + } // namespace armnn -- cgit v1.2.1