diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2021-03-12 16:29:03 +0000 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2021-03-22 22:04:20 +0000 |
commit | 4441d94fa0a97d4137e49315d69d32fdc0bbcd03 (patch) | |
tree | 3cd0482c2acb9849a9242a5699e2282857d0b0c4 | |
parent | 5b448a31940bca17911fadd48801bf7c9fe9d984 (diff) | |
download | armnn-4441d94fa0a97d4137e49315d69d32fdc0bbcd03.tar.gz |
Revert "Fix deserializer output binding tensorshape logic"
This reverts commit 4c10dfc213a59ec7bbf56b645e313083638b81a4.
Fix for IVGCVSW-5768
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I7197067a7bb407ff9d35059289a31912592d3af4
-rw-r--r-- | src/armnnDeserializer/Deserializer.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/armnnDeserializer/Deserializer.cpp b/src/armnnDeserializer/Deserializer.cpp index 8f2fa3f150..c34797725f 100644 --- a/src/armnnDeserializer/Deserializer.cpp +++ b/src/armnnDeserializer/Deserializer.cpp @@ -990,7 +990,7 @@ void IDeserializer::DeserializerImpl::SetupOutputLayers(GraphPtr graph) // GetBindingLayerInfo expect the index to be index in the vector not index property on each layer base LayerBindingId bindingId = GetBindingLayerInfo(graph, outputLayerIndex); - ARMNN_ASSERT_MSG(baseLayer->layerName()->c_str(), "Output has no name."); + ARMNN_ASSERT_MSG(baseLayer->layerName()->c_str(), "Input has no name."); IConnectableLayer* outputLayer = m_Network->AddOutputLayer(bindingId, baseLayer->layerName()->c_str()); @@ -1000,7 +1000,7 @@ void IDeserializer::DeserializerImpl::SetupOutputLayers(GraphPtr graph) unsigned int sourceLayerIndex = GetLayerIndexInVector(graph, baseLayer->inputSlots()->Get(0)->connection()->sourceLayerIndex()); LayerBaseRawPtr sourceBaseLayer = GetBaseLayer(graph, sourceLayerIndex); - const armnn::TensorInfo& tensorInfo = ToTensorInfo(sourceBaseLayer->outputSlots()->Get(i)->tensorInfo()); + const armnn::TensorInfo& tensorInfo = ToTensorInfo(sourceBaseLayer->outputSlots()->Get(0)->tensorInfo()); BindingPointInfo bindingInfo = {bindingId, tensorInfo}; m_OutputBindings.push_back(std::make_pair(baseLayer->layerName()->c_str(), bindingInfo)); |