diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-05-10 17:19:58 +0100 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-05-10 17:19:58 +0100 |
commit | cd05f3e6208d5296ae507036fe4f6df55362c5a0 (patch) | |
tree | 320056f65282bc9076935913460803f8f31a56d1 /src/armnnQuantizer/QuantizationDataSet.cpp | |
parent | bdf2aab823c123bb9d91e584cd53778f9b0524d1 (diff) | |
download | armnn-cd05f3e6208d5296ae507036fe4f6df55362c5a0.tar.gz |
IVGCVSW-2834 Fix Input TensorInfo in Quantization DataSet
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I694124aa15ef2fd368d5e6e36da8b942ae270850
Diffstat (limited to 'src/armnnQuantizer/QuantizationDataSet.cpp')
-rw-r--r-- | src/armnnQuantizer/QuantizationDataSet.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
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) |