1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5
6 #include "../Lstm.hpp"
7
8 #include <boost/test/data/test_case.hpp>
9
10 BOOST_AUTO_TEST_SUITE(LstmTests)
11
12 using namespace armnn_driver;
13
BOOST_DATA_TEST_CASE(LstmNoCifgNoPeepholeNoProjectionTest,COMPUTE_DEVICES)14 BOOST_DATA_TEST_CASE(LstmNoCifgNoPeepholeNoProjectionTest, COMPUTE_DEVICES)
15 {
16 LstmNoCifgNoPeepholeNoProjection<hal_1_2::HalPolicy>(sample);
17 }
18
BOOST_DATA_TEST_CASE(LstmCifgPeepholeNoProjectionTest,COMPUTE_DEVICES)19 BOOST_DATA_TEST_CASE(LstmCifgPeepholeNoProjectionTest, COMPUTE_DEVICES)
20 {
21 LstmCifgPeepholeNoProjection<hal_1_2::HalPolicy>(sample);
22 }
23
BOOST_DATA_TEST_CASE(LstmNoCifgPeepholeProjectionTest,COMPUTE_DEVICES)24 BOOST_DATA_TEST_CASE(LstmNoCifgPeepholeProjectionTest, COMPUTE_DEVICES)
25 {
26 LstmNoCifgPeepholeProjection<hal_1_2::HalPolicy>(sample);
27 }
28
BOOST_DATA_TEST_CASE(LstmCifgPeepholeNoProjectionBatch2Test,COMPUTE_DEVICES)29 BOOST_DATA_TEST_CASE(LstmCifgPeepholeNoProjectionBatch2Test, COMPUTE_DEVICES)
30 {
31 LstmCifgPeepholeNoProjectionBatch2<hal_1_2::HalPolicy>(sample);
32 }
33
BOOST_DATA_TEST_CASE(LstmNoCifgPeepholeProjectionNoClippingLayerNormTest,COMPUTE_DEVICES)34 BOOST_DATA_TEST_CASE(LstmNoCifgPeepholeProjectionNoClippingLayerNormTest, COMPUTE_DEVICES)
35 {
36 LstmNoCifgPeepholeProjectionNoClippingLayerNorm<hal_1_2::HalPolicy>(sample);
37 }
38
BOOST_DATA_TEST_CASE(LstmCifgPeepholeProjectionNoClippingLayerNormTest,COMPUTE_DEVICES)39 BOOST_DATA_TEST_CASE(LstmCifgPeepholeProjectionNoClippingLayerNormTest, COMPUTE_DEVICES)
40 {
41 LstmCifgPeepholeProjectionNoClippingLayerNorm<hal_1_2::HalPolicy>(sample);
42 }
43
44 #if defined(ARMCOMPUTECL_ENABLED)
BOOST_DATA_TEST_CASE(QuantizedLstmTest,COMPUTE_DEVICES)45 BOOST_DATA_TEST_CASE(QuantizedLstmTest, COMPUTE_DEVICES)
46 {
47 QuantizedLstm<hal_1_2::HalPolicy>(sample);
48 }
49 #endif
50
51 BOOST_AUTO_TEST_SUITE_END()
52