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/reference/test | |
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/reference/test')
-rw-r--r-- | src/backends/reference/test/RefLayerTests.cpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/backends/reference/test/RefLayerTests.cpp b/src/backends/reference/test/RefLayerTests.cpp index eb2d0c55d9..09096b4646 100644 --- a/src/backends/reference/test/RefLayerTests.cpp +++ b/src/backends/reference/test/RefLayerTests.cpp @@ -1283,6 +1283,8 @@ ARMNN_AUTO_TEST_CASE(LstmLayerInt16NoCifgNoPeepholeNoProjectionInt16Constant, // QLstm ARMNN_AUTO_TEST_CASE(QLstm, QLstmTest) +ARMNN_AUTO_TEST_CASE(QLstm1, QLstmTest1) +ARMNN_AUTO_TEST_CASE(QLstm2, QLstmTest2) // Convert from BFloat16 to Float32 ARMNN_AUTO_TEST_CASE(ConvertBf16ToFp32, ConvertBf16ToFp32Test) |