diff options
Diffstat (limited to 'src/armnn/LoadedNetwork.cpp')
-rw-r--r-- | src/armnn/LoadedNetwork.cpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/armnn/LoadedNetwork.cpp b/src/armnn/LoadedNetwork.cpp index 48a3040b23..fd7279a294 100644 --- a/src/armnn/LoadedNetwork.cpp +++ b/src/armnn/LoadedNetwork.cpp @@ -317,9 +317,11 @@ LoadedNetwork::LoadedNetwork(std::unique_ptr<IOptimizedNetwork> net, unsigned int inputSlotIndex = 0; for (auto& inputSlot : layer->GetInputSlots()) { - if (inputSlot.GetOwningLayer().GetType() == LayerType::Input) + if (inputSlot.GetConnectedOutputSlot()->GetOwningLayer().GetType() == LayerType::Input) { - auto inputLayer = PolymorphicDowncast<InputLayer*>(&inputSlot.GetOwningLayer()); + auto inputLayer = + PolymorphicDowncast<InputLayer*>( + &inputSlot.GetConnectedOutputSlot()->GetOwningLayer()); m_InputWorkloadSlotPairs[inputLayer->GetBindingId()] = std::make_pair(m_WorkloadQueue.size(), inputSlotIndex); } |