diff options
-rw-r--r-- | delegate/src/DelegateUtils.hpp | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/delegate/src/DelegateUtils.hpp b/delegate/src/DelegateUtils.hpp index fa38ee3e90..5dea567761 100644 --- a/delegate/src/DelegateUtils.hpp +++ b/delegate/src/DelegateUtils.hpp @@ -587,8 +587,7 @@ TfLiteStatus ProcessInputs(armnn::IConnectableLayer* layer, armnn::IOutputSlot& outputSlot = constantLayer->GetOutputSlot(0); outputSlot.SetTensorInfo(inputTensorInfo); - delegateData.m_OutputSlotForNode[static_cast<unsigned long>(inputIndex)] = &outputSlot; - + delegateData.m_OutputSlotForNode[tfLiteNode->inputs->data[inputIndex]] = &outputSlot; } } |