diff options
author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-05-07 15:01:00 +0100 |
---|---|---|
committer | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-05-07 15:01:00 +0100 |
commit | 59f32f96c0fc5c451e0fdbbfc24b072b07dc226b (patch) | |
tree | 5477b29b70fc3785316f9600ffc8bce414e4440c /src/armnn/QuantizerVisitor.cpp | |
parent | eb2b329b761ce3206505ed8d2eab071a2f97d5e7 (diff) | |
download | armnn-59f32f96c0fc5c451e0fdbbfc24b072b07dc226b.tar.gz |
IVGCVSW-3067 Fix issue with QuantizerVisior not setting TensorShape correctly
* Also update unit tests for the quantizer to check the input and output
shape of the network
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Change-Id: Ie804d852f2a4770d98dfb4dab6415f3d0fdd2ce4
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); |