From cd05f3e6208d5296ae507036fe4f6df55362c5a0 Mon Sep 17 00:00:00 2001 From: Narumol Prangnawarat Date: Fri, 10 May 2019 17:19:58 +0100 Subject: IVGCVSW-2834 Fix Input TensorInfo in Quantization DataSet Signed-off-by: Narumol Prangnawarat Change-Id: I694124aa15ef2fd368d5e6e36da8b942ae270850 --- src/armnnQuantizer/QuantizationDataSet.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/armnnQuantizer/QuantizationDataSet.cpp b/src/armnnQuantizer/QuantizationDataSet.cpp index d225883854..ac819dc133 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) -- cgit v1.2.1