From b22a75e2aaec1175bbacba54e1a33a83f9749ce2 Mon Sep 17 00:00:00 2001 From: James Conroy Date: Mon, 8 Jun 2020 14:53:10 +0100 Subject: 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 Change-Id: I93d7c64c4a9cc1012cb2bc052d598d4279fbd372 --- src/backends/reference/test/RefLayerTests.cpp | 2 ++ 1 file changed, 2 insertions(+) (limited to 'src/backends/reference/test/RefLayerTests.cpp') 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) -- cgit v1.2.1