diff options
author | James Conroy <james.conroy@arm.com> | 2019-08-01 16:01:48 +0100 |
---|---|---|
committer | James Conroy <james.conroy@arm.com> | 2019-08-02 12:37:04 +0100 |
commit | 9c3cae8683e4b24932446b88d3ecbc02f9f9fa08 (patch) | |
tree | ccfd7ca42d51594c7ff2bc908e4b27e8601b8a90 /src/backends/backendsCommon/test/LayerTests.hpp | |
parent | 61c6da2f467c5c2a3a1748b551a1274d238f12bb (diff) | |
download | armnn-9c3cae8683e4b24932446b88d3ecbc02f9f9fa08.tar.gz |
IVGCVSW-3470 Add Quantized_LSTM tests
* Added Layer and Create Workload tests
for the new Quantized LSTM layer.
* Tests to be enabled on NEON and CL in
their respective patches.
Signed-off-by: James Conroy <james.conroy@arm.com>
Change-Id: I7e9e9768dd63010ab58367c45fffcff452377cfb
Diffstat (limited to 'src/backends/backendsCommon/test/LayerTests.hpp')
-rw-r--r-- | src/backends/backendsCommon/test/LayerTests.hpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/test/LayerTests.hpp b/src/backends/backendsCommon/test/LayerTests.hpp index fb7ce92702..1121ae71e0 100644 --- a/src/backends/backendsCommon/test/LayerTests.hpp +++ b/src/backends/backendsCommon/test/LayerTests.hpp @@ -1536,6 +1536,11 @@ LayerTestResult<int16_t, 2> LstmLayerInt16NoCifgNoPeepholeNoProjectionInt16Const armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); +// QuantizedLstm +LayerTestResult<uint8_t, 2> QuantizedLstmTest( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); + LayerTestResult<float, 4> SimpleConvertFp16ToFp32Test( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); |