aboutsummaryrefslogtreecommitdiff
path: root/ConversionUtils.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'ConversionUtils.cpp')
-rw-r--r--ConversionUtils.cpp9
1 files changed, 9 insertions, 0 deletions
diff --git a/ConversionUtils.cpp b/ConversionUtils.cpp
index 4c773964..b03ffbda 100644
--- a/ConversionUtils.cpp
+++ b/ConversionUtils.cpp
@@ -38,6 +38,15 @@ void LayerInputHandle::Connect(armnn::IInputSlot& inputSlot)
}
}
+void LayerInputHandle::Disconnect(armnn::IInputSlot& inputSlot)
+{
+ ARMNN_ASSERT(IsValid());
+ if (m_OutputSlot)
+ {
+ m_OutputSlot->Disconnect(inputSlot);
+ }
+}
+
const armnn::TensorInfo& LayerInputHandle::GetTensorInfo() const
{
return m_TensorInfo;