diff options
Diffstat (limited to 'src/armnn/DynamicQuantizationVisitor.cpp')
-rw-r--r-- | src/armnn/DynamicQuantizationVisitor.cpp | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/src/armnn/DynamicQuantizationVisitor.cpp b/src/armnn/DynamicQuantizationVisitor.cpp index ba87c6d335..4b1dce0b6f 100644 --- a/src/armnn/DynamicQuantizationVisitor.cpp +++ b/src/armnn/DynamicQuantizationVisitor.cpp @@ -63,13 +63,14 @@ void DynamicQuantizationVisitor::RemoveDebugLayers() for (DebugLayer* debugLayer : m_DebugLayers) { OutputSlot& proceedingOutputSlot = *debugLayer->GetInputSlot(0).GetConnectedOutputSlot(); - InputSlot& succeedingInputSlot = *debugLayer->GetOutputSlot(0).GetConnection(0); proceedingOutputSlot.Disconnect(debugLayer->GetInputSlot(0)); - debugLayer->GetOutputSlot(0).Disconnect(succeedingInputSlot); + for (InputSlot* succeedingInputSlot : debugLayer->GetOutputSlot(0).GetConnections()) + { + debugLayer->GetOutputSlot(0).Disconnect(*succeedingInputSlot); + proceedingOutputSlot.Connect(*succeedingInputSlot); + } m_Graph.EraseLayer(debugLayer); - - proceedingOutputSlot.Connect(succeedingInputSlot); } m_DebugLayers.clear(); } |