3 files changed, 14 insertions, 0 deletions
diff --git a/include/armnn/INetwork.hpp b/include/armnn/INetwork.hpp
index 475367ece5..1ed560040f 100644
--- a/include/armnn/INetwork.hpp
+++ b/include/armnn/INetwork.hpp
@@ -29,6 +29,7 @@ public:
     virtual const IOutputSlot* GetConnection() const = 0;
     virtual IOutputSlot* GetConnection() = 0;
     virtual const IConnectableLayer& GetOwningIConnectableLayer() const = 0;
+    virtual IConnectableLayer& GetOwningIConnectableLayer() = 0;
 
 protected:
    /// Not user deletable.
@@ -56,6 +57,7 @@ public:
     virtual LayerGuid GetOwningLayerGuid() const = 0;
 
     virtual const IConnectableLayer& GetOwningIConnectableLayer() const = 0;
+    virtual IConnectableLayer& GetOwningIConnectableLayer() = 0;
 
 protected:
     /// Not user deletable.
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
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;