diff options
author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-04-01 17:04:53 +0100 |
---|---|---|
committer | Ruomei Yan <ruomei.yan@arm.com> | 2019-04-02 10:00:17 +0000 |
commit | a0beb3b8aeb8bbea906609e0f50a250c33cde10b (patch) | |
tree | 543352916c2424bef670619dcdd0f7f78e754678 /src/backends/backendsCommon/test/LayerTests.cpp | |
parent | 2ab0bfa335896cc5f514732e181b0bcceb44b141 (diff) | |
download | armnn-a0beb3b8aeb8bbea906609e0f50a250c33cde10b.tar.gz |
IVGCVSW-2872 Unit tests for Quantize layer and reference workload
Change-Id: I291c08cb6e359453978b398255cf8ff051ed2686
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Diffstat (limited to 'src/backends/backendsCommon/test/LayerTests.cpp')
-rw-r--r-- | src/backends/backendsCommon/test/LayerTests.cpp | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/test/LayerTests.cpp b/src/backends/backendsCommon/test/LayerTests.cpp index be1d43541c..e8fb84f534 100644 --- a/src/backends/backendsCommon/test/LayerTests.cpp +++ b/src/backends/backendsCommon/test/LayerTests.cpp @@ -44,6 +44,7 @@ #include "ConvertFp32ToFp16TestImpl.hpp" #include "DebugTestImpl.hpp" #include "DequantizeTestImpl.hpp" +#include "QuantizeTestImpl.hpp" // 3-channel 16x8 image used as common input data for a number of Conv2d tests. static std::vector<float> ConvInput3x8x16({ @@ -8977,3 +8978,24 @@ LayerTestResult<float, 4> DequantizeUint8Test( { return DequantizeSimpleTest<armnn::DataType::QuantisedAsymm8>(workloadFactory, memoryManager); } + +LayerTestResult<uint8_t, 4> QuantizeSimpleUint8Test( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) +{ + return QuantizeSimpleTest<armnn::DataType::QuantisedAsymm8>(workloadFactory, memoryManager); +} + +LayerTestResult<uint8_t, 4> QuantizeClampUint8Test( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) +{ + return QuantizeClampTest<armnn::DataType::QuantisedAsymm8>(workloadFactory, memoryManager); +} + +LayerTestResult<int16_t, 4> QuantizeClampInt16Test( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) +{ + return QuantizeClampTest<armnn::DataType::QuantisedSymm16>(workloadFactory, memoryManager); +} |