diff options
Diffstat (limited to 'test/1.2/UnidirectionalSequenceLstm.cpp')
-rw-r--r-- | test/1.2/UnidirectionalSequenceLstm.cpp | 40 |
1 files changed, 40 insertions, 0 deletions
diff --git a/test/1.2/UnidirectionalSequenceLstm.cpp b/test/1.2/UnidirectionalSequenceLstm.cpp new file mode 100644 index 00000000..fd35aa41 --- /dev/null +++ b/test/1.2/UnidirectionalSequenceLstm.cpp @@ -0,0 +1,40 @@ +// +// Copyright © 2022 Arm Ltd and Contributors. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#include "../UnidirectionalSequenceLstm.hpp" + +using namespace armnn_driver; + +DOCTEST_TEST_SUITE("UnidirectionalSequenceLstmTests_1.2_CpuRef") +{ + + DOCTEST_TEST_CASE("UnidirectionalSequenceLstmLayerFloat32Test_1.2_CpuRef") + { + UnidirectionalSequenceLstmLayerFloat32TestImpl<hal_1_2::HalPolicy>(armnn::Compute::CpuRef); + } + + DOCTEST_TEST_CASE("UnidirectionalSequenceLstmLayerFloat32TimeMajorTest_1.2_CpuRef") + { + UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl<hal_1_2::HalPolicy>(armnn::Compute::CpuRef); + } + + DOCTEST_TEST_CASE("UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest_1.2_CpuRef") + { + UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTestImpl<hal_1_2::HalPolicy> + (armnn::Compute::CpuRef); + } + + DOCTEST_TEST_CASE("UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest_1.2_CpuRef") + { + UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl<hal_1_2::HalPolicy> + (armnn::Compute::CpuRef); + } + + DOCTEST_TEST_CASE("UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest_1.2_CpuRef") + { + UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTestImpl<hal_1_2::HalPolicy>(armnn::Compute::CpuRef); + } + +}
\ No newline at end of file |