commit 0fa5e6dda7e9e76d7ca5f77777c35374c768f28c
author Cathal Corbett <cathal.corbett@arm.com> Fri Jan 21 16:55:13 2022 +0000
committer Cathal Corbett <cathal.corbett@arm.com> Mon Jan 24 18:10:44 2022 +0000
tree fe4cdfe345398429328be5e5485cea5d3a0073be
parent e02562b0957b4c14739f608513d3c03050c19369

IVGCVSW-6269 Add support of Unidirectional Sequence Lstm fp32/fp16 to Android

  * Unidirectional Sequence Lstm is supported in android-nn-driver.
  * CTS and VTS pass for Unidirectional Sequence Lstm if there are
    any without weights as inputs. If that isn't the case use the
    tests implemented in the android driver for LSTM.

Signed-off-by: Cathal Corbett <cathal.corbett@arm.com>
Change-Id: Ie3a237b67f5101c4f9a2c7bea796e9c674bedba1

test/1.2/UnidirectionalSequenceLstm.cpp

diff --git a/test/1.2/UnidirectionalSequenceLstm.cpp b/test/1.2/UnidirectionalSequenceLstm.cpp
new file mode 100644
index 0000000..fd35aa4
--- /dev/null
+++ b/test/1.2/UnidirectionalSequenceLstm.cpp
@@ -0,0 +1,40 @@
+//
+// Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#include "../UnidirectionalSequenceLstm.hpp"
+
+using namespace armnn_driver;
+
+DOCTEST_TEST_SUITE("UnidirectionalSequenceLstmTests_1.2_CpuRef")
+{
+
+    DOCTEST_TEST_CASE("UnidirectionalSequenceLstmLayerFloat32Test_1.2_CpuRef")
+    {
+        UnidirectionalSequenceLstmLayerFloat32TestImpl<hal_1_2::HalPolicy>(armnn::Compute::CpuRef);
+    }
+
+    DOCTEST_TEST_CASE("UnidirectionalSequenceLstmLayerFloat32TimeMajorTest_1.2_CpuRef")
+    {
+        UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl<hal_1_2::HalPolicy>(armnn::Compute::CpuRef);
+    }
+
+    DOCTEST_TEST_CASE("UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest_1.2_CpuRef")
+    {
+        UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTestImpl<hal_1_2::HalPolicy>
+            (armnn::Compute::CpuRef);
+    }
+
+    DOCTEST_TEST_CASE("UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest_1.2_CpuRef")
+    {
+        UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl<hal_1_2::HalPolicy>
+            (armnn::Compute::CpuRef);
+    }
+
+    DOCTEST_TEST_CASE("UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest_1.2_CpuRef")
+    {
+        UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTestImpl<hal_1_2::HalPolicy>(armnn::Compute::CpuRef);
+    }
+
+}
\ No newline at end of file