diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2021-05-27 09:55:43 +0100 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2021-05-27 11:35:54 +0000 |
commit | bf99b5f05514d3a717df36c6039dcfc4a9f5b9ba (patch) | |
tree | 8fc934576760db1e10b7dbc8bffbd4e7002bc733 /src/armnnTfLiteParser/TfLiteParser.cpp | |
parent | f1c7f0020874ada342cacf301b1d137e67751deb (diff) | |
download | armnn-bf99b5f05514d3a717df36c6039dcfc4a9f5b9ba.tar.gz |
IVGCVSW-6079 Fix circular dependency
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I29793ece7b6bfc015c643be3ed16529ab50f0d7d
Diffstat (limited to 'src/armnnTfLiteParser/TfLiteParser.cpp')
-rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.cpp | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp index d4a0a6e865..8941ee93f5 100644 --- a/src/armnnTfLiteParser/TfLiteParser.cpp +++ b/src/armnnTfLiteParser/TfLiteParser.cpp @@ -1963,10 +1963,9 @@ void TfLiteParserImpl::ParsePrelu(size_t subgraphIndex, size_t operatorIndex) 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); |