diff options
author | Jan Eilers <jan.eilers@arm.com> | 2019-07-08 09:57:55 +0100 |
---|---|---|
committer | Jan Eilers <jan.eilers@arm.com> | 2019-07-10 09:15:04 +0000 |
commit | ad5293a86e315049de36afd723dcd1a7e70681a7 (patch) | |
tree | b9003cd1fba00c267a971d899284b3fcbd5ce6f5 /src/backends/neon/test/NeonLayerTests.cpp | |
parent | 8b797a84f1e8f9d1d5d064afbc4fc12c21b8ffed (diff) | |
download | armnn-ad5293a86e315049de36afd723dcd1a7e70681a7.tar.gz |
IVGCVSW-3337 Add Neon backend support for LSTM layer normalisation
* Update neon lstm workload
* Add unit tests
* Add isLstmSupported
Change-Id: I493c159137f6544b0f2532d16d4fafd7a7e587e5
Signed-off-by: Jan Eilers <jan.eilers@arm.com>
Diffstat (limited to 'src/backends/neon/test/NeonLayerTests.cpp')
-rw-r--r-- | src/backends/neon/test/NeonLayerTests.cpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/backends/neon/test/NeonLayerTests.cpp b/src/backends/neon/test/NeonLayerTests.cpp index 51fd219365..049680aafe 100644 --- a/src/backends/neon/test/NeonLayerTests.cpp +++ b/src/backends/neon/test/NeonLayerTests.cpp @@ -469,6 +469,8 @@ ARMNN_AUTO_TEST_CASE(LstmLayerFloat32NoCifgNoPeepholeNoProjection, LstmLayerFloat32NoCifgNoPeepholeNoProjectionTest) ARMNN_AUTO_TEST_CASE(LstmLayerFloat32NoCifgWithPeepholeWithProjection, LstmLayerFloat32NoCifgWithPeepholeWithProjectionTest) +ARMNN_AUTO_TEST_CASE(LstmLayerFloat32NoCifgWithPeepholeWithProjectionWithLayerNorm, + LstmLayerFloat32NoCifgWithPeepholeWithProjectionWithLayerNormTest) // Mean ARMNN_AUTO_TEST_CASE(MeanSimpleFloat32, MeanSimpleTest<armnn::DataType::Float32>) |