diff options
Diffstat (limited to 'src/armnn/Layer.cpp')
-rw-r--r-- | src/armnn/Layer.cpp | 13 |
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; |