diff options
Diffstat (limited to 'test/1.0/Lstm.cpp')
-rw-r--r-- | test/1.0/Lstm.cpp | 34 |
1 files changed, 34 insertions, 0 deletions
diff --git a/test/1.0/Lstm.cpp b/test/1.0/Lstm.cpp new file mode 100644 index 00000000..5f0a209d --- /dev/null +++ b/test/1.0/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_0::HalPolicy>(sample); +} + +BOOST_DATA_TEST_CASE(LstmCifgPeepholeNoProjectionTest, COMPUTE_DEVICES) +{ + LstmCifgPeepholeNoProjection<hal_1_0::HalPolicy>(sample); +} + +BOOST_DATA_TEST_CASE(LstmNoCifgPeepholeProjectionTest, COMPUTE_DEVICES) +{ + LstmNoCifgPeepholeProjection<hal_1_0::HalPolicy>(sample); +} + +BOOST_DATA_TEST_CASE(LstmCifgPeepholeNoProjectionBatch2Test, COMPUTE_DEVICES) +{ + LstmCifgPeepholeNoProjectionBatch2<hal_1_0::HalPolicy>(sample); +} + +BOOST_AUTO_TEST_SUITE_END() |