diff options
author | Jan Eilers <jan.eilers@arm.com> | 2019-07-08 15:56:59 +0100 |
---|---|---|
committer | Nikhil Raj Arm <nikhil.raj@arm.com> | 2019-07-09 15:08:10 +0000 |
commit | a2ec9092f0bff018bfe7ae0cacb7e30bcc17c1c7 (patch) | |
tree | 11a36098492aed1629d873dcc02fb25a0071de2b /src/backends/cl/test/ClLayerTests.cpp | |
parent | c0ed7baa8c05c4710034dfd179fadd31b716a46f (diff) | |
download | armnn-a2ec9092f0bff018bfe7ae0cacb7e30bcc17c1c7.tar.gz |
IVGCVSW-3338 Add CL backend support for LSTM normalization
* Enable calls to LSTM normalization unit tests on CL backend.
* Update CL workload to set the layer normalization parameters.
!android-nn-driver:1461
Change-Id: Ia5a29918961c391c1f1d8f331add377a38822ddd
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>
Signed-off-by: Jan Eilers <jan.eilers@arm.com>
Diffstat (limited to 'src/backends/cl/test/ClLayerTests.cpp')
-rw-r--r-- | src/backends/cl/test/ClLayerTests.cpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/backends/cl/test/ClLayerTests.cpp b/src/backends/cl/test/ClLayerTests.cpp index ac96bf8135..5575a05b99 100644 --- a/src/backends/cl/test/ClLayerTests.cpp +++ b/src/backends/cl/test/ClLayerTests.cpp @@ -354,6 +354,9 @@ ARMNN_AUTO_TEST_CASE(LstmLayerFloat32NoCifgNoPeepholeNoProjection, ARMNN_AUTO_TEST_CASE(LstmLayerFloat32NoCifgWithPeepholeWithProjection, LstmLayerFloat32NoCifgWithPeepholeWithProjectionTest) +ARMNN_AUTO_TEST_CASE(LstmLayerFloat32NoCifgWithPeepholeWithProjectionWithLayerNorm, + LstmLayerFloat32NoCifgWithPeepholeWithProjectionWithLayerNormTest) + // Convert from Float16 to Float32 ARMNN_AUTO_TEST_CASE(SimpleConvertFp16ToFp32, SimpleConvertFp16ToFp32Test) // Convert from Float32 to Float16 |