commit bf99b5f05514d3a717df36c6039dcfc4a9f5b9ba
author Narumol Prangnawarat <narumol.prangnawarat@arm.com> Thu May 27 09:55:43 2021 +0100
committer Jim Flynn <jim.flynn@arm.com> Thu May 27 11:35:54 2021 +0000
tree 8fc934576760db1e10b7dbc8bffbd4e7002bc733
parent f1c7f0020874ada342cacf301b1d137e67751deb

IVGCVSW-6079 Fix circular dependency

Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I29793ece7b6bfc015c643be3ed16529ab50f0d7d

src/armnnTfLiteParser/TfLiteParser.cpp

diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp
index d4a0a6e..8941ee9 100644
--- a/src/armnnTfLiteParser/TfLiteParser.cpp
+++ b/src/armnnTfLiteParser/TfLiteParser.cpp
@@ -1963,10 +1963,9 @@
 
     if (IsConstTensor(inputs[1]))
     {
-        armnn::IInputSlot* slot = &(layer->GetInputSlot(0));
-        RegisterConsumerOfTensor(subgraphIndex, 0, slot);
-
         auto inputTensorIndexes = AsUnsignedVector(GetInputTensorIds(m_Model, subgraphIndex, operatorIndex));
+        armnn::IInputSlot* slot = &(layer->GetInputSlot(0));
+        RegisterConsumerOfTensor(subgraphIndex, inputTensorIndexes[0], slot);
 
         auto alphaTensorAndData = CreateConstTensorNonPermuted(inputs[1], alphaTensorInfo);
         std::string constLayerName = fmt::format("Constant:{}", inputs[1]->name);