commit 4441d94fa0a97d4137e49315d69d32fdc0bbcd03
author Teresa Charlin <teresa.charlinreyes@arm.com> Fri Mar 12 16:29:03 2021 +0000
committer Jim Flynn <jim.flynn@arm.com> Mon Mar 22 22:04:20 2021 +0000
tree 3cd0482c2acb9849a9242a5699e2282857d0b0c4
parent 5b448a31940bca17911fadd48801bf7c9fe9d984

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

src/armnnDeserializer/Deserializer.cpp

diff --git a/src/armnnDeserializer/Deserializer.cpp b/src/armnnDeserializer/Deserializer.cpp
index 8f2fa3f..c347977 100644
--- a/src/armnnDeserializer/Deserializer.cpp
+++ b/src/armnnDeserializer/Deserializer.cpp
@@ -990,7 +990,7 @@
 
         // 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 @@
         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));