diff options
| author | James Conroy <james.conroy@arm.com> | 2020-06-08 14:53:10 +0100 |
|---|---|---|
| committer | James Conroy <james.conroy@arm.com> | 2020-06-08 15:15:42 +0100 |
| commit | b22a75e2aaec1175bbacba54e1a33a83f9749ce2 (patch) | |
| tree | e25c0ba8652435a14825eb29a7847a421efadb84 /src/backends/backendsCommon/test/layerTests/LstmTestImpl.hpp | |
| parent | 041b3c0d416d94b1aa4efa41960808b7e5e61f25 (diff) | |
| download | armnn-b22a75e2aaec1175bbacba54e1a33a83f9749ce2.tar.gz | |
IVGCVSW-4860 Add tests to verify QLstm projection
* Adds int16 output tensor to CpuRef impl to
prevent overflow when accumulating output
after projection.
* Adds two remaining tests to verify QLstm on
CpuRef.
Signed-off-by: James Conroy <james.conroy@arm.com>
Change-Id: I93d7c64c4a9cc1012cb2bc052d598d4279fbd372
Diffstat (limited to 'src/backends/backendsCommon/test/layerTests/LstmTestImpl.hpp')
| -rw-r--r-- | src/backends/backendsCommon/test/layerTests/LstmTestImpl.hpp | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/test/layerTests/LstmTestImpl.hpp b/src/backends/backendsCommon/test/layerTests/LstmTestImpl.hpp index f1180aee16..6e293456af 100644 --- a/src/backends/backendsCommon/test/layerTests/LstmTestImpl.hpp +++ b/src/backends/backendsCommon/test/layerTests/LstmTestImpl.hpp @@ -66,3 +66,11 @@ LayerTestResult<uint8_t, 2> QuantizedLstmTest( LayerTestResult<int8_t, 2> QLstmTest( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); + +LayerTestResult<int8_t, 2> QLstmTest1( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); + +LayerTestResult<int8_t, 2> QLstmTest2( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
\ No newline at end of file |