aboutsummaryrefslogtreecommitdiff
path: root/test/1.1/Lstm.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'test/1.1/Lstm.cpp')
-rw-r--r--test/1.1/Lstm.cpp34
1 files changed, 34 insertions, 0 deletions
diff --git a/test/1.1/Lstm.cpp b/test/1.1/Lstm.cpp
new file mode 100644
index 00000000..703597e5
--- /dev/null
+++ b/test/1.1/Lstm.cpp
@@ -0,0 +1,34 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#include "../Lstm.hpp"
+
+#include <boost/test/data/test_case.hpp>
+
+BOOST_AUTO_TEST_SUITE(LstmTests)
+
+using namespace armnn_driver;
+
+BOOST_DATA_TEST_CASE(LstmNoCifgNoPeepholeNoProjectionTest, COMPUTE_DEVICES)
+{
+ LstmNoCifgNoPeepholeNoProjection<hal_1_1::HalPolicy>(sample);
+}
+
+BOOST_DATA_TEST_CASE(LstmCifgPeepholeNoProjectionTest, COMPUTE_DEVICES)
+{
+ LstmCifgPeepholeNoProjection<hal_1_1::HalPolicy>(sample);
+}
+
+BOOST_DATA_TEST_CASE(LstmNoCifgPeepholeProjectionTest, COMPUTE_DEVICES)
+{
+ LstmNoCifgPeepholeProjection<hal_1_1::HalPolicy>(sample);
+}
+
+BOOST_DATA_TEST_CASE(LstmCifgPeepholeNoProjectionBatch2Test, COMPUTE_DEVICES)
+{
+ LstmCifgPeepholeNoProjectionBatch2<hal_1_1::HalPolicy>(sample);
+}
+
+BOOST_AUTO_TEST_SUITE_END()