commit cd05f3e6208d5296ae507036fe4f6df55362c5a0
author Narumol Prangnawarat <narumol.prangnawarat@arm.com> Fri May 10 17:19:58 2019 +0100
committer Narumol Prangnawarat <narumol.prangnawarat@arm.com> Fri May 10 17:19:58 2019 +0100
tree 320056f65282bc9076935913460803f8f31a56d1
parent bdf2aab823c123bb9d91e584cd53778f9b0524d1

IVGCVSW-2834 Fix Input TensorInfo in Quantization DataSet

Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I694124aa15ef2fd368d5e6e36da8b942ae270850

src/armnnQuantizer/QuantizationDataSet.cpp

diff --git a/src/armnnQuantizer/QuantizationDataSet.cpp b/src/armnnQuantizer/QuantizationDataSet.cpp
index d225883..ac819dc 100644
--- a/src/armnnQuantizer/QuantizationDataSet.cpp
+++ b/src/armnnQuantizer/QuantizationDataSet.cpp
@@ -51,7 +51,7 @@
                                         armnn::LayerBindingId id,
                                         const char* name)
 {
-    m_TensorInfos.emplace(id, layer->GetInputSlot(0).GetConnection()->GetTensorInfo());
+    m_TensorInfos.emplace(id, layer->GetOutputSlot(0).GetTensorInfo());
 }
 
 armnn::TensorInfo InputLayerVisitor::GetTensorInfo(armnn::LayerBindingId layerBindingId)