diff options
| author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-10-29 17:58:36 +0000 |
|---|---|---|
| committer | Áron Virginás-Tar <aron.virginas-tar@arm.com> | 2019-10-30 13:41:10 +0000 |
| commit | c0a87c14bd2bc8a02f6c5c9f919abca27ca4dde0 (patch) | |
| tree | 3bdd47db9baec3b1566eee83145866946913f8bf /src/armnn/test/TensorTest.cpp | |
| parent | c04019985db1ee44c71834892ad17365185a3f8d (diff) | |
| download | armnn-c0a87c14bd2bc8a02f6c5c9f919abca27ca4dde0.tar.gz | |
IVGCVSW-3831 Add support of per-axis quantization to TensorInfo
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: Iea09539c92d51e546fbad8b2903b59fc08d66618
Diffstat (limited to 'src/armnn/test/TensorTest.cpp')
| -rw-r--r-- | src/armnn/test/TensorTest.cpp | 34 |
1 files changed, 34 insertions, 0 deletions
diff --git a/src/armnn/test/TensorTest.cpp b/src/armnn/test/TensorTest.cpp index a0a6c7e91f..154a0bca04 100644 --- a/src/armnn/test/TensorTest.cpp +++ b/src/armnn/test/TensorTest.cpp @@ -143,4 +143,38 @@ BOOST_AUTO_TEST_CASE(TensorShapeOperatorBrackets) BOOST_TEST(shape[2] == 20); } +BOOST_AUTO_TEST_CASE(TensorInfoPerAxisQuantization) +{ + // Old constructor + TensorInfo tensorInfo0({ 1, 1 }, DataType::Float32, 2.0f, 1); + BOOST_CHECK(!tensorInfo0.HasMultipleQuantizationScales()); + BOOST_CHECK(tensorInfo0.GetQuantizationScale() == 2.0f); + BOOST_CHECK(tensorInfo0.GetQuantizationOffset() == 1); + BOOST_CHECK(tensorInfo0.GetQuantizationScales()[0] == 2.0f); + BOOST_CHECK(!tensorInfo0.GetQuantizationDim().has_value()); + + // Set per-axis quantization scales + std::vector<float> perAxisScales{ 3.0f, 4.0f }; + tensorInfo0.SetQuantizationScales(perAxisScales); + BOOST_CHECK(tensorInfo0.HasMultipleQuantizationScales()); + BOOST_CHECK(tensorInfo0.GetQuantizationScales() == perAxisScales); + + // Set per-tensor quantization scale + tensorInfo0.SetQuantizationScale(5.0f); + BOOST_CHECK(!tensorInfo0.HasMultipleQuantizationScales()); + BOOST_CHECK(tensorInfo0.GetQuantizationScales()[0] == 5.0f); + + // Set quantization offset + tensorInfo0.SetQuantizationDim(Optional<unsigned int>(1)); + BOOST_CHECK(tensorInfo0.GetQuantizationDim().value() == 1); + + // New constructor + perAxisScales = { 6.0f, 7.0f }; + TensorInfo tensorInfo1({ 1, 1 }, DataType::Float32, perAxisScales, 1); + BOOST_CHECK(tensorInfo1.HasMultipleQuantizationScales()); + BOOST_CHECK(tensorInfo1.GetQuantizationOffset() == 0); + BOOST_CHECK(tensorInfo1.GetQuantizationScales() == perAxisScales); + BOOST_CHECK(tensorInfo1.GetQuantizationDim().value() == 1); +} + BOOST_AUTO_TEST_SUITE_END() |