diff options
Diffstat (limited to 'test/1.2/Lstm.cpp')
-rw-r--r-- | test/1.2/Lstm.cpp | 44 |
1 files changed, 44 insertions, 0 deletions
diff --git a/test/1.2/Lstm.cpp b/test/1.2/Lstm.cpp new file mode 100644 index 00000000..a76b7610 --- /dev/null +++ b/test/1.2/Lstm.cpp @@ -0,0 +1,44 @@ +// +// 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_2::HalPolicy>(sample); +} + +BOOST_DATA_TEST_CASE(LstmCifgPeepholeNoProjectionTest, COMPUTE_DEVICES) +{ + LstmCifgPeepholeNoProjection<hal_1_2::HalPolicy>(sample); +} + +BOOST_DATA_TEST_CASE(LstmNoCifgPeepholeProjectionTest, COMPUTE_DEVICES) +{ + LstmNoCifgPeepholeProjection<hal_1_2::HalPolicy>(sample); +} + +BOOST_DATA_TEST_CASE(LstmCifgPeepholeNoProjectionBatch2Test, COMPUTE_DEVICES) +{ + LstmCifgPeepholeNoProjectionBatch2<hal_1_2::HalPolicy>(sample); +} + +BOOST_DATA_TEST_CASE(LstmNoCifgPeepholeProjectionNoClippingLayerNormTest, COMPUTE_DEVICES) +{ + LstmNoCifgPeepholeProjectionNoClippingLayerNorm<hal_1_2::HalPolicy>(sample); +} + +BOOST_DATA_TEST_CASE(LstmCifgPeepholeProjectionNoClippingLayerNormTest, COMPUTE_DEVICES) +{ + LstmCifgPeepholeProjectionNoClippingLayerNorm<hal_1_2::HalPolicy>(sample); +} + +BOOST_AUTO_TEST_SUITE_END() |