diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2024-02-01 17:36:48 +0000 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2024-02-07 13:24:09 +0000 |
commit | a52bca23d225144e92f521341718a70489d5c217 (patch) | |
tree | 2dc88b168900ec89e51dafe564856736ce88f277 /src/backends/reference/test/RefEndToEndTests.cpp | |
parent | cf9622f84e22dfb4af1835f02780eb0d8310a624 (diff) | |
download | armnn-a52bca23d225144e92f521341718a70489d5c217.tar.gz |
IVGCVSW-7623: GpuFsa Op: Add Pool2d operator
* Add Pool2d EndToEnd tests to all backends
* Add utility functions for the attributes in a separate file
* Remove some unnecessary includes
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I0f82ebbf7b3301c6368462fb4fb4d4d02b246fc6
Diffstat (limited to 'src/backends/reference/test/RefEndToEndTests.cpp')
-rw-r--r-- | src/backends/reference/test/RefEndToEndTests.cpp | 46 |
1 files changed, 45 insertions, 1 deletions
diff --git a/src/backends/reference/test/RefEndToEndTests.cpp b/src/backends/reference/test/RefEndToEndTests.cpp index c09304e4d1..9f80059712 100644 --- a/src/backends/reference/test/RefEndToEndTests.cpp +++ b/src/backends/reference/test/RefEndToEndTests.cpp @@ -28,6 +28,7 @@ #include <backendsCommon/test/GatherNdEndToEndTestImpl.hpp> #include <backendsCommon/test/InstanceNormalizationEndToEndTestImpl.hpp> #include <backendsCommon/test/LogSoftmaxEndToEndTestImpl.hpp> +#include "backendsCommon/test/Pooling2dEndToEndTestImpl.hpp" #include <backendsCommon/test/PreluEndToEndTestImpl.hpp> #include <backendsCommon/test/QLstmEndToEndTestImpl.hpp> #include <backendsCommon/test/QuantizationEndToEndTestImpl.hpp> @@ -1098,6 +1099,49 @@ TEST_CASE("RefPreluEndToEndTestQSymm16") PreluEndToEndPositiveTest<armnn::DataType::QSymmS16>(defaultBackends); } +// Pooling 2D +// Average Pool 2D +TEST_CASE("RefAvgPool2DEndtoEndTestFloat32") +{ + AvgPool2dEndToEnd<DataType::Float32>(defaultBackends); +} + +TEST_CASE("RefAvgPool2DEndtoEndTestFloat16") +{ + AvgPool2dEndToEndFloat16<DataType::Float16>(defaultBackends); +} + +TEST_CASE("RefAvgPool2DIgnoreValueEndtoEndTestFloat32") +{ + AvgPool2dEndToEnd<DataType::Float32>(defaultBackends, PaddingMethod::IgnoreValue); +} + +// Max Pool 2D +TEST_CASE("RefMaxPool2DEndtoEndTestFloat32") +{ + MaxPool2dEndToEnd<DataType::Float32>(defaultBackends); +} + +TEST_CASE("RefMaxPool2DEndtoEndTestFloat16") +{ + MaxPool2dEndToEndFloat16<DataType::Float16>(defaultBackends); +} + +TEST_CASE("RefMaxPool2DIgnoreValueEndtoEndTestFloat32") +{ + MaxPool2dEndToEnd<DataType::Float32>(defaultBackends, PaddingMethod::IgnoreValue); +} + +TEST_CASE("RefMaxPool2DTwoLayerEndtoEndTestFloat32") +{ + MaxPool2dTwoLayerEndToEnd<DataType::Float32>(defaultBackends); +} + +TEST_CASE("RefMaxPool2DThreeLayerEndtoEndTestFloat32") +{ + MaxPool2dThreeLayerEndToEnd<DataType::Float32>(defaultBackends); +} + // Quantization TEST_CASE("QuantizationEndToEndFloat32_U8Test") { @@ -1156,7 +1200,7 @@ TEST_CASE("RefSplit1dEndtoEndTestSigned16") Splitter1dEndToEnd<DataType::QSymmS16>(defaultBackends); } -TEST_CASE("TosaRefSplit1dEndtoEndTestFloat16") +TEST_CASE("RefSplit1dEndtoEndTestFloat16") { Splitter1dEndToEndFloat16<DataType::Float16>(defaultBackends); } |