Ferran Balaguer | b2397fd | 2019-07-25 12:12:39 +0100 | [diff] [blame] | 1 | // |
Mike Kelly | e2d611e | 2021-10-14 12:35:58 +0100 | [diff] [blame^] | 2 | // Copyright © 2017 Arm Ltd and Contributors. All rights reserved. |
Ferran Balaguer | b2397fd | 2019-07-25 12:12:39 +0100 | [diff] [blame] | 3 | // SPDX-License-Identifier: MIT |
| 4 | // |
| 5 | |
| 6 | #include "../Lstm.hpp" |
| 7 | |
Ferran Balaguer | b2397fd | 2019-07-25 12:12:39 +0100 | [diff] [blame] | 8 | using namespace armnn_driver; |
| 9 | |
Mike Kelly | e2d611e | 2021-10-14 12:35:58 +0100 | [diff] [blame^] | 10 | DOCTEST_TEST_SUITE("LstmTests_1.2_CpuRef") |
Ferran Balaguer | b2397fd | 2019-07-25 12:12:39 +0100 | [diff] [blame] | 11 | { |
Mike Kelly | e2d611e | 2021-10-14 12:35:58 +0100 | [diff] [blame^] | 12 | |
| 13 | DOCTEST_TEST_CASE("LstmNoCifgNoPeepholeNoProjectionTest_1.2_armnn::Compute::CpuRef") |
Sadik Armagan | 9150bff | 2021-05-26 15:40:53 +0100 | [diff] [blame] | 14 | { |
| 15 | LstmNoCifgNoPeepholeNoProjection<hal_1_2::HalPolicy>(armnn::Compute::CpuRef); |
| 16 | } |
Mike Kelly | e2d611e | 2021-10-14 12:35:58 +0100 | [diff] [blame^] | 17 | |
| 18 | DOCTEST_TEST_CASE("LstmCifgPeepholeNoProjectionTest_1.2_CpuRef") |
Sadik Armagan | 9150bff | 2021-05-26 15:40:53 +0100 | [diff] [blame] | 19 | { |
| 20 | LstmCifgPeepholeNoProjection<hal_1_2::HalPolicy>(armnn::Compute::CpuRef); |
| 21 | } |
Mike Kelly | e2d611e | 2021-10-14 12:35:58 +0100 | [diff] [blame^] | 22 | |
| 23 | DOCTEST_TEST_CASE("LstmNoCifgPeepholeProjectionTest_1.2_CpuRef") |
Sadik Armagan | 9150bff | 2021-05-26 15:40:53 +0100 | [diff] [blame] | 24 | { |
| 25 | LstmNoCifgPeepholeProjection<hal_1_2::HalPolicy>(armnn::Compute::CpuRef); |
| 26 | } |
Mike Kelly | e2d611e | 2021-10-14 12:35:58 +0100 | [diff] [blame^] | 27 | |
| 28 | DOCTEST_TEST_CASE("LstmCifgPeepholeNoProjectionBatch2Test_1.2_CpuRef") |
Sadik Armagan | 9150bff | 2021-05-26 15:40:53 +0100 | [diff] [blame] | 29 | { |
| 30 | LstmCifgPeepholeNoProjectionBatch2<hal_1_2::HalPolicy>(armnn::Compute::CpuRef); |
| 31 | } |
Mike Kelly | e2d611e | 2021-10-14 12:35:58 +0100 | [diff] [blame^] | 32 | |
| 33 | DOCTEST_TEST_CASE("QuantizedLstmTest_1.2_CpuRef") |
Sadik Armagan | 9150bff | 2021-05-26 15:40:53 +0100 | [diff] [blame] | 34 | { |
| 35 | QuantizedLstm<hal_1_2::HalPolicy>(armnn::Compute::CpuRef); |
| 36 | } |
Mike Kelly | e2d611e | 2021-10-14 12:35:58 +0100 | [diff] [blame^] | 37 | |
Ferran Balaguer | b2397fd | 2019-07-25 12:12:39 +0100 | [diff] [blame] | 38 | } |
| 39 | |
David Monahan | c60d0fd | 2020-05-19 14:58:34 +0100 | [diff] [blame] | 40 | #if defined(ARMCOMPUTECL_ENABLED) |
Mike Kelly | e2d611e | 2021-10-14 12:35:58 +0100 | [diff] [blame^] | 41 | DOCTEST_TEST_SUITE("LstmTests_1.2_GpuAcc") |
Ellen Norris-Thompson | a3d7fad | 2019-08-05 14:20:32 +0100 | [diff] [blame] | 42 | { |
Mike Kelly | e2d611e | 2021-10-14 12:35:58 +0100 | [diff] [blame^] | 43 | |
| 44 | DOCTEST_TEST_CASE("LstmNoCifgNoPeepholeNoProjectionTest_1.2_GpuAcc") |
Sadik Armagan | 9150bff | 2021-05-26 15:40:53 +0100 | [diff] [blame] | 45 | { |
| 46 | LstmNoCifgNoPeepholeNoProjection<hal_1_2::HalPolicy>(armnn::Compute::GpuAcc); |
| 47 | } |
Mike Kelly | e2d611e | 2021-10-14 12:35:58 +0100 | [diff] [blame^] | 48 | |
| 49 | DOCTEST_TEST_CASE("LstmCifgPeepholeNoProjectionTest_1.2_GpuAcc") |
Sadik Armagan | 9150bff | 2021-05-26 15:40:53 +0100 | [diff] [blame] | 50 | { |
| 51 | LstmCifgPeepholeNoProjection<hal_1_2::HalPolicy>(armnn::Compute::GpuAcc); |
| 52 | } |
Mike Kelly | e2d611e | 2021-10-14 12:35:58 +0100 | [diff] [blame^] | 53 | |
| 54 | DOCTEST_TEST_CASE("LstmNoCifgPeepholeProjectionTest_1.2_GpuAcc") |
Sadik Armagan | 9150bff | 2021-05-26 15:40:53 +0100 | [diff] [blame] | 55 | { |
| 56 | LstmNoCifgPeepholeProjection<hal_1_2::HalPolicy>(armnn::Compute::GpuAcc); |
| 57 | } |
Mike Kelly | e2d611e | 2021-10-14 12:35:58 +0100 | [diff] [blame^] | 58 | |
| 59 | DOCTEST_TEST_CASE("LstmCifgPeepholeNoProjectionBatch2Test_1.2_GpuAcc") |
Sadik Armagan | 9150bff | 2021-05-26 15:40:53 +0100 | [diff] [blame] | 60 | { |
| 61 | LstmCifgPeepholeNoProjectionBatch2<hal_1_2::HalPolicy>(armnn::Compute::GpuAcc); |
| 62 | } |
Mike Kelly | e2d611e | 2021-10-14 12:35:58 +0100 | [diff] [blame^] | 63 | |
| 64 | DOCTEST_TEST_CASE("QuantizedLstmTest_1.2_GpuAcc") |
Sadik Armagan | 9150bff | 2021-05-26 15:40:53 +0100 | [diff] [blame] | 65 | { |
| 66 | QuantizedLstm<hal_1_2::HalPolicy>(armnn::Compute::GpuAcc); |
| 67 | } |
Mike Kelly | e2d611e | 2021-10-14 12:35:58 +0100 | [diff] [blame^] | 68 | |
Ellen Norris-Thompson | a3d7fad | 2019-08-05 14:20:32 +0100 | [diff] [blame] | 69 | } |
David Monahan | c60d0fd | 2020-05-19 14:58:34 +0100 | [diff] [blame] | 70 | #endif |