aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/optimizations/RedirectMembersToConstantInputs.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/optimizations/RedirectMembersToConstantInputs.hpp')
-rw-r--r--src/armnn/optimizations/RedirectMembersToConstantInputs.hpp2
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.