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.1_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.1_armnn::Compute::CpuRef") |
Sadik Armagan | 9150bff | 2021-05-26 15:40:53 +0100 | [diff] [blame] | 14 | { |
| 15 | LstmNoCifgNoPeepholeNoProjection<hal_1_1::HalPolicy>(armnn::Compute::CpuRef); |
| 16 | } |
Mike Kelly | e2d611e | 2021-10-14 12:35:58 +0100 | [diff] [blame^] | 17 | |
| 18 | DOCTEST_TEST_CASE("LstmCifgPeepholeNoProjectionTest_1.1_CpuRef") |
Sadik Armagan | 9150bff | 2021-05-26 15:40:53 +0100 | [diff] [blame] | 19 | { |
| 20 | LstmCifgPeepholeNoProjection<hal_1_1::HalPolicy>(armnn::Compute::CpuRef); |
| 21 | } |
Mike Kelly | e2d611e | 2021-10-14 12:35:58 +0100 | [diff] [blame^] | 22 | |
| 23 | DOCTEST_TEST_CASE("LstmNoCifgPeepholeProjectionTest_1.1_CpuRef") |
Sadik Armagan | 9150bff | 2021-05-26 15:40:53 +0100 | [diff] [blame] | 24 | { |
| 25 | LstmNoCifgPeepholeProjection<hal_1_1::HalPolicy>(armnn::Compute::CpuRef); |
| 26 | } |
Mike Kelly | e2d611e | 2021-10-14 12:35:58 +0100 | [diff] [blame^] | 27 | |
| 28 | DOCTEST_TEST_CASE("LstmCifgPeepholeNoProjectionBatch2Test_1.1_CpuRef") |
Sadik Armagan | 9150bff | 2021-05-26 15:40:53 +0100 | [diff] [blame] | 29 | { |
| 30 | LstmCifgPeepholeNoProjectionBatch2<hal_1_1::HalPolicy>(armnn::Compute::CpuRef); |
| 31 | } |
Mike Kelly | e2d611e | 2021-10-14 12:35:58 +0100 | [diff] [blame^] | 32 | |
Ferran Balaguer | b2397fd | 2019-07-25 12:12:39 +0100 | [diff] [blame] | 33 | } |
| 34 | |
Sadik Armagan | 9150bff | 2021-05-26 15:40:53 +0100 | [diff] [blame] | 35 | #if defined(ARMCOMPUTECL_ENABLED) |
Mike Kelly | e2d611e | 2021-10-14 12:35:58 +0100 | [diff] [blame^] | 36 | DOCTEST_TEST_SUITE("LstmTests_1.1_GpuAcc") |
Ferran Balaguer | b2397fd | 2019-07-25 12:12:39 +0100 | [diff] [blame] | 37 | { |
Mike Kelly | e2d611e | 2021-10-14 12:35:58 +0100 | [diff] [blame^] | 38 | |
| 39 | DOCTEST_TEST_CASE("LstmNoCifgNoPeepholeNoProjectionTest_1.1_GpuAcc") |
Sadik Armagan | 9150bff | 2021-05-26 15:40:53 +0100 | [diff] [blame] | 40 | { |
| 41 | LstmNoCifgNoPeepholeNoProjection<hal_1_1::HalPolicy>(armnn::Compute::GpuAcc); |
| 42 | } |
Mike Kelly | e2d611e | 2021-10-14 12:35:58 +0100 | [diff] [blame^] | 43 | |
| 44 | DOCTEST_TEST_CASE("LstmCifgPeepholeNoProjectionTest_1.1_GpuAcc") |
Sadik Armagan | 9150bff | 2021-05-26 15:40:53 +0100 | [diff] [blame] | 45 | { |
| 46 | LstmCifgPeepholeNoProjection<hal_1_1::HalPolicy>(armnn::Compute::GpuAcc); |
| 47 | } |
Mike Kelly | e2d611e | 2021-10-14 12:35:58 +0100 | [diff] [blame^] | 48 | |
| 49 | DOCTEST_TEST_CASE("LstmNoCifgPeepholeProjectionTest_1.1_GpuAcc") |
Sadik Armagan | 9150bff | 2021-05-26 15:40:53 +0100 | [diff] [blame] | 50 | { |
| 51 | LstmNoCifgPeepholeProjection<hal_1_1::HalPolicy>(armnn::Compute::GpuAcc); |
| 52 | } |
Mike Kelly | e2d611e | 2021-10-14 12:35:58 +0100 | [diff] [blame^] | 53 | |
| 54 | DOCTEST_TEST_CASE("LstmCifgPeepholeNoProjectionBatch2Test_1.1_GpuAcc") |
Sadik Armagan | 9150bff | 2021-05-26 15:40:53 +0100 | [diff] [blame] | 55 | { |
| 56 | LstmCifgPeepholeNoProjectionBatch2<hal_1_1::HalPolicy>(armnn::Compute::GpuAcc); |
| 57 | } |
Mike Kelly | e2d611e | 2021-10-14 12:35:58 +0100 | [diff] [blame^] | 58 | |
Ferran Balaguer | b2397fd | 2019-07-25 12:12:39 +0100 | [diff] [blame] | 59 | } |
Sadik Armagan | 9150bff | 2021-05-26 15:40:53 +0100 | [diff] [blame] | 60 | #endif |