From bd575b270f65601ff7bdfdc58de45b9675d5541a Mon Sep 17 00:00:00 2001 From: Narumol Prangnawarat Date: Tue, 31 Aug 2021 16:53:54 +0100 Subject: MLCE-530 Add support of int8 weight for UnidirectionalSequenceLstm to Ref backend and armnn delegate Signed-off-by: Narumol Prangnawarat Change-Id: I203d0029c12221228ffe229acda3c90594394e9b --- .../UnidirectionalSequenceLstmTestImpl.hpp | 25 ++++++++++++++++++++++ 1 file changed, 25 insertions(+) (limited to 'src/backends/backendsCommon/test/layerTests/UnidirectionalSequenceLstmTestImpl.hpp') diff --git a/src/backends/backendsCommon/test/layerTests/UnidirectionalSequenceLstmTestImpl.hpp b/src/backends/backendsCommon/test/layerTests/UnidirectionalSequenceLstmTestImpl.hpp index 7b14065728..20ac3135a4 100644 --- a/src/backends/backendsCommon/test/layerTests/UnidirectionalSequenceLstmTestImpl.hpp +++ b/src/backends/backendsCommon/test/layerTests/UnidirectionalSequenceLstmTestImpl.hpp @@ -31,6 +31,31 @@ LayerTestResult UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithP const armnn::ITensorHandleFactory& tensorHandleFactory); LayerTestResult UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ITensorHandleFactory& tensorHandleFactory); + +LayerTestResult UnidirectionalSequenceLstmLayerInt8Test( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ITensorHandleFactory& tensorHandleFactory); + +LayerTestResult UnidirectionalSequenceLstmLayerInt8TimeMajorTest( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ITensorHandleFactory& tensorHandleFactory); + +LayerTestResult UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ITensorHandleFactory& tensorHandleFactory); + +LayerTestResult UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ITensorHandleFactory& tensorHandleFactory); + +LayerTestResult UnidirectionalSequenceLstmInt8WithCifgWithPeepholeNoProjectionTest( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, const armnn::ITensorHandleFactory& tensorHandleFactory); \ No newline at end of file -- cgit v1.2.1