1 // 2 // Copyright © 2017 Arm Ltd and Contributors. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #include "../Lstm.hpp" 7 8 using namespace armnn_driver; 9 10 #if defined(ARMNNREF_ENABLED) 11 DOCTEST_TEST_SUITE("LstmTests_1.2_CpuRef") 12 { 13 14 DOCTEST_TEST_CASE("LstmNoCifgNoPeepholeNoProjectionTest_1.2_armnn::Compute::CpuRef") 15 { 16 LstmNoCifgNoPeepholeNoProjection<hal_1_2::HalPolicy>(armnn::Compute::CpuRef); 17 } 18 19 DOCTEST_TEST_CASE("LstmCifgPeepholeNoProjectionTest_1.2_CpuRef") 20 { 21 LstmCifgPeepholeNoProjection<hal_1_2::HalPolicy>(armnn::Compute::CpuRef); 22 } 23 24 DOCTEST_TEST_CASE("LstmNoCifgPeepholeProjectionTest_1.2_CpuRef") 25 { 26 LstmNoCifgPeepholeProjection<hal_1_2::HalPolicy>(armnn::Compute::CpuRef); 27 } 28 29 DOCTEST_TEST_CASE("LstmCifgPeepholeNoProjectionBatch2Test_1.2_CpuRef") 30 { 31 LstmCifgPeepholeNoProjectionBatch2<hal_1_2::HalPolicy>(armnn::Compute::CpuRef); 32 } 33 34 DOCTEST_TEST_CASE("QuantizedLstmTest_1.2_CpuRef") 35 { 36 QuantizedLstm<hal_1_2::HalPolicy>(armnn::Compute::CpuRef); 37 } 38 39 } 40 #endif 41 42 #if defined(ARMCOMPUTECL_ENABLED) 43 DOCTEST_TEST_SUITE("LstmTests_1.2_GpuAcc") 44 { 45 46 DOCTEST_TEST_CASE("LstmNoCifgNoPeepholeNoProjectionTest_1.2_GpuAcc") 47 { 48 LstmNoCifgNoPeepholeNoProjection<hal_1_2::HalPolicy>(armnn::Compute::GpuAcc); 49 } 50 51 DOCTEST_TEST_CASE("LstmCifgPeepholeNoProjectionTest_1.2_GpuAcc") 52 { 53 LstmCifgPeepholeNoProjection<hal_1_2::HalPolicy>(armnn::Compute::GpuAcc); 54 } 55 56 DOCTEST_TEST_CASE("LstmNoCifgPeepholeProjectionTest_1.2_GpuAcc") 57 { 58 LstmNoCifgPeepholeProjection<hal_1_2::HalPolicy>(armnn::Compute::GpuAcc); 59 } 60 61 DOCTEST_TEST_CASE("LstmCifgPeepholeNoProjectionBatch2Test_1.2_GpuAcc") 62 { 63 LstmCifgPeepholeNoProjectionBatch2<hal_1_2::HalPolicy>(armnn::Compute::GpuAcc); 64 } 65 66 DOCTEST_TEST_CASE("QuantizedLstmTest_1.2_GpuAcc") 67 { 68 QuantizedLstm<hal_1_2::HalPolicy>(armnn::Compute::GpuAcc); 69 } 70 71 } 72 #endif 73