aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/Layer.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/Layer.cpp')
-rw-r--r--src/armnn/Layer.cpp13
1 files changed, 11 insertions, 2 deletions
diff --git a/src/armnn/Layer.cpp b/src/armnn/Layer.cpp
index 0a6328ba3d..ced87b095c 100644
--- a/src/armnn/Layer.cpp
+++ b/src/armnn/Layer.cpp
@@ -262,8 +262,17 @@ LayerPriority Layer::GetPriority() const
auto maxPrio = [](const LayerPriority prio, const InputSlot& slot) -> LayerPriority
{
- const Layer& input = slot.GetConnectedOutputSlot()->GetOwningLayer();
- return std::max(prio, input.GetPriority());
+ const OutputSlot *outputSlot = slot.GetConnectedOutputSlot();
+ if (outputSlot)
+ {
+ const Layer& input = outputSlot->GetOwningLayer();
+ return std::max(prio, input.GetPriority());
+ }
+ else
+ {
+ // unconnected input slot
+ return prio;
+ }
};
m_Visiting = true;