// // Copyright © 2017 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #include "../Lstm.hpp" using namespace armnn_driver; DOCTEST_TEST_SUITE("LstmTests_1.0_CpuRef") { DOCTEST_TEST_CASE("LstmNoCifgNoPeepholeNoProjectionTest_1.0_armnn::Compute::CpuRef") { LstmNoCifgNoPeepholeNoProjection(armnn::Compute::CpuRef); } DOCTEST_TEST_CASE("LstmCifgPeepholeNoProjectionTest_1.0_CpuRef") { LstmCifgPeepholeNoProjection(armnn::Compute::CpuRef); } DOCTEST_TEST_CASE("LstmNoCifgPeepholeProjectionTest_1.0_CpuRef") { LstmNoCifgPeepholeProjection(armnn::Compute::CpuRef); } DOCTEST_TEST_CASE("LstmCifgPeepholeNoProjectionBatch2Test_1.0_CpuRef") { LstmCifgPeepholeNoProjectionBatch2(armnn::Compute::CpuRef); } } #if defined(ARMCOMPUTECL_ENABLED) DOCTEST_TEST_SUITE("LstmTests_1.0_GpuAcc") { DOCTEST_TEST_CASE("LstmNoCifgNoPeepholeNoProjectionTest_1.0_GpuAcc") { LstmNoCifgNoPeepholeNoProjection(armnn::Compute::GpuAcc); } DOCTEST_TEST_CASE("LstmCifgPeepholeNoProjectionTest_1.0_GpuAcc") { LstmCifgPeepholeNoProjection(armnn::Compute::GpuAcc); } DOCTEST_TEST_CASE("LstmNoCifgPeepholeProjectionTest_1.0_GpuAcc") { LstmNoCifgPeepholeProjection(armnn::Compute::GpuAcc); } DOCTEST_TEST_CASE("LstmCifgPeepholeNoProjectionBatch2Test_1.0_GpuAcc") { LstmCifgPeepholeNoProjectionBatch2(armnn::Compute::GpuAcc); } } #endif