diff options
Diffstat (limited to 'ConversionUtils.cpp')
-rw-r--r-- | ConversionUtils.cpp | 9 |
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; |