diff options
-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) |