diff options
author | Nikhil Raj <nikraj01@e126673.cambridge.arm.com> | 2022-08-03 18:20:59 +0100 |
---|---|---|
committer | Nikhil Raj Arm <nikhil.raj@arm.com> | 2022-08-05 13:20:15 +0000 |
commit | 2803c8d09dd380b361f92dd57fc897eab5033a95 (patch) | |
tree | a5a2afed0b781337d95a1259861ea2d13acab6b0 | |
parent | 8c7c569fca0988776f108ba4a56c1f60e176b62c (diff) | |
download | armnn-2803c8d09dd380b361f92dd57fc897eab5033a95.tar.gz |
IVGCVSW-7103 ExecuteNetwork failed while doing Yolov4 execution
* Adding the check only if it's not a const layer which is needed to run ai_benchmark_v5_yolo_v4_tiny_quant.tflite model
* We still won't be able to run the model due to IVGCVSW-7158
Signed-off-by: Nikhil Raj <nikraj01@e126673.cambridge.arm.com>
Change-Id: Ib7e77a0b5a64be0c92a8e4eae45729f799770b37
-rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.cpp | 19 |
1 files changed, 11 insertions, 8 deletions
diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp index 33567953a2..880de100c1 100644 --- a/src/armnnTfLiteParser/TfLiteParser.cpp +++ b/src/armnnTfLiteParser/TfLiteParser.cpp @@ -948,16 +948,19 @@ void TfLiteParserImpl::RegisterProducerOfTensor(size_t subgraphIndex, TensorSlots & tensorSlots = m_SubgraphConnections[subgraphIndex][tensorIndex]; - // assuming there is only one producer for that tensor - if (tensorSlots.outputSlot != nullptr) + if (slot->GetOwningIConnectableLayer().GetType() != LayerType::Constant) { - throw ParseException(fmt::format("Another layer has already registered itself as the producer of " - "subgraph:{} tensor:{} {}", - subgraphIndex, - tensorIndex, - CHECK_LOCATION().AsString())); - } + // assuming there is only one producer for that tensor + if (tensorSlots.outputSlot != nullptr) + { + throw ParseException(fmt::format("Another layer has already registered itself as the producer of " + "subgraph:{} tensor:{} {}", + subgraphIndex, + tensorIndex, + CHECK_LOCATION().AsString())); + } + } tensorSlots.outputSlot = slot; } |