diff options
Diffstat (limited to 'src/armnn/optimizations/RedirectMembersToConstantInputs.hpp')
-rw-r--r-- | src/armnn/optimizations/RedirectMembersToConstantInputs.hpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/armnn/optimizations/RedirectMembersToConstantInputs.hpp b/src/armnn/optimizations/RedirectMembersToConstantInputs.hpp index 483377452e..a2bad710e6 100644 --- a/src/armnn/optimizations/RedirectMembersToConstantInputs.hpp +++ b/src/armnn/optimizations/RedirectMembersToConstantInputs.hpp @@ -63,6 +63,8 @@ private: for (unsigned int inputSlotIndex = 1; inputSlotIndex != layerPtr->GetNumInputSlots(); ++inputSlotIndex) { OutputSlot* outputSlot = layerPtr->GetInputSlot(inputSlotIndex).GetConnectedOutputSlot(); + // Debug layers should not be inserted in optimize process yet + ARMNN_ASSERT(outputSlot->GetOwningLayer().GetType() != LayerType::Debug); if (outputSlot->GetOwningLayer().GetType() == LayerType::Constant) { // Get constant layer and redirect base layer member variables. |