diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-06-18 16:22:35 +0100 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-06-18 17:29:15 +0100 |
commit | 0244cd0f70378d1a8eaebcbc04e1e76cb71be528 (patch) | |
tree | 0b2c8b37f66910825b6f2d7d51b3a66edfa8217b /src | |
parent | 71c80b1d51ce596df1d9f2c6c1b49640e7437eaa (diff) | |
download | armnn-0244cd0f70378d1a8eaebcbc04e1e76cb71be528.tar.gz |
IVGCVSW-3234 Ensure that Quantizer allows different quantization scales
on inputs and output of Concat layer
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Idd79e44dbf49345aced3ddeeb8a53c9776f9f5d5
Diffstat (limited to 'src')
-rw-r--r-- | src/armnn/test/QuantizerTest.cpp | 18 |
1 files changed, 16 insertions, 2 deletions
diff --git a/src/armnn/test/QuantizerTest.cpp b/src/armnn/test/QuantizerTest.cpp index 48602d31f7..581991b57c 100644 --- a/src/armnn/test/QuantizerTest.cpp +++ b/src/armnn/test/QuantizerTest.cpp @@ -86,6 +86,11 @@ protected: } } + void TestDifferentQuantizationScale(const TensorInfo& info0, const TensorInfo& info1) + { + BOOST_TEST(info0.GetQuantizationScale() != info1.GetQuantizationScale()); + } + void TestConstantQuantizationParams(const TensorInfo& info, const OffsetScalePair& params, DataType dataType = DataType::QuantisedAsymm8) @@ -1319,10 +1324,19 @@ BOOST_AUTO_TEST_CASE(QuantizeConcat) const OriginsDescriptor& originsDescriptor, const char* name = nullptr) override { - TensorInfo info = layer->GetOutputSlot(0).GetTensorInfo(); + TensorInfo outputInfo = layer->GetOutputSlot(0).GetTensorInfo(); TestQuantizationParams( - info, {60.8f / g_Asymm8QuantizationBase, 65}, {45.3f / g_Symm16QuantizationBase, 0}); + outputInfo, {60.8f / g_Asymm8QuantizationBase, 65}, {45.3f / g_Symm16QuantizationBase, 0}); + + TensorInfo inputInfo0 = layer->GetInputSlot(0).GetConnection()->GetTensorInfo(); + TensorInfo inputInfo1 = layer->GetInputSlot(1).GetConnection()->GetTensorInfo(); + TensorInfo inputInfo2 = layer->GetInputSlot(2).GetConnection()->GetTensorInfo(); + + TestDifferentQuantizationScale(inputInfo0, inputInfo1); + TestDifferentQuantizationScale(inputInfo0, inputInfo2); + TestDifferentQuantizationScale(inputInfo1, inputInfo2); + TestDifferentQuantizationScale(inputInfo0, outputInfo); } }; |