diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2021-05-07 14:16:13 +0100 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2021-05-07 14:54:28 +0000 |
commit | 67ac7fac3453fbeaa146a5b52f688a5b804296c2 (patch) | |
tree | 9cb8ea60b799b84d3c0dafff9e1dfd962b15230b /delegate/src | |
parent | 27dee45706c666d8d1fe66cafe5cea81e68b28bf (diff) | |
download | armnn-67ac7fac3453fbeaa146a5b52f688a5b804296c2.tar.gz |
IVGCVSW-5418 'ExecuteNetwork test for MobileBERT'
* Set the correct index for connecting constant layers.
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I86a626b82943e0d04b8ecb6b4f0f1b750b22d161
Diffstat (limited to 'delegate/src')
-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; } } |