diff options
Diffstat (limited to 'src/armnn/QuantizerVisitor.cpp')
-rw-r--r-- | src/armnn/QuantizerVisitor.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/armnn/QuantizerVisitor.cpp b/src/armnn/QuantizerVisitor.cpp index 95f7c50735..919eda1c7e 100644 --- a/src/armnn/QuantizerVisitor.cpp +++ b/src/armnn/QuantizerVisitor.cpp @@ -49,7 +49,7 @@ void QuantizerVisitor::SetQuantizedInputConnections(const IConnectableLayer* src OffsetScalePair qParams = m_QuantizationScheme->ComputeScheme(range.first, range.second); // Set the quantization params - TensorInfo info(newOutputSlot.GetTensorInfo()); + TensorInfo info(outputSlot->GetTensorInfo()); info.SetDataType(m_QuantizationScheme->GetDataType()); info.SetQuantizationOffset(qParams.second); info.SetQuantizationScale(qParams.first); |