From: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Date: Fri, 10 May 2019 16:19:58 +0000 (+0100)
Subject: IVGCVSW-2834 Fix Input TensorInfo in Quantization DataSet
X-Git-Tag: submit/tizen/20200316.035456~652
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=cd05f3e6208d5296ae507036fe4f6df55362c5a0;p=platform%2Fupstream%2Farmnn.git

IVGCVSW-2834 Fix Input TensorInfo in Quantization DataSet

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

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 @@ void InputLayerVisitor::VisitInputLayer(const armnn::IConnectableLayer* layer,
                                         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)