diff options
author | Josh Minor <josh.minor@arm.com> | 2020-06-17 13:56:20 -0500 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2020-06-23 15:48:07 +0000 |
commit | 4c10dfc213a59ec7bbf56b645e313083638b81a4 (patch) | |
tree | 696bb45fc7940a44808432978128db48341567a5 /src/armnnDeserializer/Deserializer.cpp | |
parent | 8b23635b3049b24dc424905f33345456f3f6eb7d (diff) | |
download | armnn-4c10dfc213a59ec7bbf56b645e313083638b81a4.tar.gz |
Fix deserializer output binding tensorshape logic
Signed-off-by: Josh Minor <josh.minor@arm.com>
Change-Id: I090733fb7b04ac6841a802eeec0fa663f72da9a2
Diffstat (limited to 'src/armnnDeserializer/Deserializer.cpp')
-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 3b69ed17b8..b9579d563d 100644 --- a/src/armnnDeserializer/Deserializer.cpp +++ b/src/armnnDeserializer/Deserializer.cpp @@ -925,7 +925,7 @@ void Deserializer::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(), "Input has no name."); + ARMNN_ASSERT_MSG(baseLayer->layerName()->c_str(), "Output has no name."); IConnectableLayer* outputLayer = m_Network->AddOutputLayer(bindingId, baseLayer->layerName()->c_str()); @@ -935,7 +935,7 @@ void Deserializer::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(0)->tensorInfo()); + const armnn::TensorInfo& tensorInfo = ToTensorInfo(sourceBaseLayer->outputSlots()->Get(i)->tensorInfo()); BindingPointInfo bindingInfo = {bindingId, tensorInfo}; m_OutputBindings.push_back(std::make_pair(baseLayer->layerName()->c_str(), bindingInfo)); |