diff options
| author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2021-08-31 16:53:54 +0100 |
|---|---|---|
| committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2021-08-31 16:53:54 +0100 |
| commit | bd575b270f65601ff7bdfdc58de45b9675d5541a (patch) | |
| tree | 0063b2d9716f5d1bab45577992a6e193522912d6 /src/backends/reference | |
| parent | 7684b18e8fec45355a49e7f7165c582efc553ab6 (diff) | |
| download | armnn-bd575b270f65601ff7bdfdc58de45b9675d5541a.tar.gz | |
MLCE-530 Add support of int8 weight for UnidirectionalSequenceLstm
to Ref backend and armnn delegate
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I203d0029c12221228ffe229acda3c90594394e9b
Diffstat (limited to 'src/backends/reference')
| -rw-r--r-- | src/backends/reference/RefLayerSupport.cpp | 5 | ||||
| -rw-r--r-- | src/backends/reference/test/RefLayerTests.cpp | 10 |
2 files changed, 13 insertions, 2 deletions
diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp index 2603371927..5eba3e5919 100644 --- a/src/backends/reference/RefLayerSupport.cpp +++ b/src/backends/reference/RefLayerSupport.cpp @@ -2317,9 +2317,10 @@ bool RefLayerSupport::IsUnidirectionalSequenceLstmSupported( DataType::Float32 }; - std::array<DataType, 1> supportedWeightTypes = + std::array<DataType, 2> supportedWeightTypes = { - DataType::Float32 + DataType::Float32, + DataType::QAsymmS8 }; // check inputs and outputs diff --git a/src/backends/reference/test/RefLayerTests.cpp b/src/backends/reference/test/RefLayerTests.cpp index 0cf36f2c6e..e906b2962c 100644 --- a/src/backends/reference/test/RefLayerTests.cpp +++ b/src/backends/reference/test/RefLayerTests.cpp @@ -2341,5 +2341,15 @@ ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmLayerNoCifgWithPeepholeW UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest) ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjection, UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest) +ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmLayerInt8, + UnidirectionalSequenceLstmLayerInt8Test) +ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmLayerInt8TimeMajor, + UnidirectionalSequenceLstmLayerInt8TimeMajorTest) +ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjection, + UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest) +ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNorm, + UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest) +ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmInt8WithCifgWithPeepholeNoProjection, + UnidirectionalSequenceLstmInt8WithCifgWithPeepholeNoProjectionTest) }
\ No newline at end of file |