diff options
author | Nikhil Raj <nikraj01@e126673.cambridge.arm.com> | 2022-08-03 18:20:59 +0100 |
---|---|---|
committer | Nikhil Raj <nikhil.raj@arm.com> | 2022-08-05 16:33:54 +0100 |
commit | d4d1c31c077d3b73b625465a00926ac16ad90f73 (patch) | |
tree | a5a2afed0b781337d95a1259861ea2d13acab6b0 /src/armnnTfLiteParser/TfLiteParser.cpp | |
parent | 210897d7afb4007fe06d390c2ec333a732390b5c (diff) | |
download | armnn-d4d1c31c077d3b73b625465a00926ac16ad90f73.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
Diffstat (limited to 'src/armnnTfLiteParser/TfLiteParser.cpp')
-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; } |