diff options
author | Ferran Balaguer <ferran.balaguer@arm.com> | 2019-07-25 12:12:39 +0100 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-07-29 13:55:18 +0000 |
commit | b2397fd9d68f12792b04dd90ae72dfb8cb9f4a60 (patch) | |
tree | 69833c482fd57df20e478398abdf932205bf99bb /test/1.1 | |
parent | a3609ccd88a9c403e086e88cad063adedbcad5fd (diff) | |
download | android-nn-driver-b2397fd9d68f12792b04dd90ae72dfb8cb9f4a60.tar.gz |
IVGCVSW-3398 Add LSTM normalization parameters to HAL 1.2
* Adding LSTM processing function in HAL 1.2 with normalization parameters
* Refactoring LSTM tests
!armnn:1608
Signed-off-by: Ferran Balaguer <ferran.balaguer@arm.com>
Change-Id: I0e00f14ef078a333e9f2f23d6278a5d92a3001d6
Diffstat (limited to 'test/1.1')
-rw-r--r-- | test/1.1/Lstm.cpp | 34 |
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() |