From 7ac07f355f4cb75a54ec423670b7078bd0ecb44d Mon Sep 17 00:00:00 2001 From: Nattapat Chaimanowong Date: Wed, 20 Mar 2019 11:51:14 +0000 Subject: IVGCVSW-2858 Add support for QSymm16 quantization Change-Id: Ia7c305c30c39ec0e9db447a461479be17fde250c Signed-off-by: Nattapat Chaimanowong --- src/armnn/test/QuantizerTest.cpp | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) (limited to 'src/armnn/test') diff --git a/src/armnn/test/QuantizerTest.cpp b/src/armnn/test/QuantizerTest.cpp index f7723bd0c0..eead9b7f36 100644 --- a/src/armnn/test/QuantizerTest.cpp +++ b/src/armnn/test/QuantizerTest.cpp @@ -9,8 +9,9 @@ #include #include "armnn/LayerVisitorBase.hpp" -#include "../Network.hpp" #include "../Graph.hpp" +#include "../Network.hpp" +#include "../NetworkQuantizationScheme.hpp" #include "../NetworkQuantizerUtils.hpp" #include "../OverrideInputRangeVisitor.hpp" #include "../RangeTracker.hpp" @@ -997,7 +998,9 @@ BOOST_AUTO_TEST_CASE(QuantizeMerger) const OriginsDescriptor& mergerDescriptor, const char* name = nullptr) { - std::pair expectedValues = ComputeQAsymmParams(8, m_Min, m_Max); + + QAsymm8QuantizationScheme quantizationScheme; + OffsetScalePair expectedValues = quantizationScheme.ComputeScheme(m_Min, m_Max); TensorInfo info = layer->GetOutputSlot(0).GetTensorInfo(); -- cgit v1.2.1