diff options
| author | Sadik Armagan <sadik.armagan@arm.com> | 2021-05-07 14:16:13 +0100 |
|---|---|---|
| committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2021-05-07 18:12:54 +0100 |
| commit | f370e312bd26c29c454ea82bb08ddb614f010484 (patch) | |
| tree | c2aaadcbe987885a3ed5629f36759b1ff9c62c86 | |
| parent | 69703a558c3f09f40b6b8387f47ba7e09e118036 (diff) | |
| download | armnn-f370e312bd26c29c454ea82bb08ddb614f010484.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
| -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; } } |