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/cl/test | |
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/cl/test')
-rw-r--r-- | src/backends/cl/test/ClEndToEndTests.cpp | 52 |
1 files changed, 48 insertions, 4 deletions
diff --git a/src/backends/cl/test/ClEndToEndTests.cpp b/src/backends/cl/test/ClEndToEndTests.cpp index 878054f7ba..3acd7dc29b 100644 --- a/src/backends/cl/test/ClEndToEndTests.cpp +++ b/src/backends/cl/test/ClEndToEndTests.cpp @@ -17,6 +17,7 @@ #include <backendsCommon/test/ElementwiseUnaryEndToEndTestImpl.hpp> #include <backendsCommon/test/FillEndToEndTestImpl.hpp> #include <backendsCommon/test/InstanceNormalizationEndToEndTestImpl.hpp> +#include "backendsCommon/test/Pooling2dEndToEndTestImpl.hpp" #include <backendsCommon/test/PreluEndToEndTestImpl.hpp> #include <backendsCommon/test/QLstmEndToEndTestImpl.hpp> #include <backendsCommon/test/QuantizedLstmEndToEndTestImpl.hpp> @@ -245,8 +246,8 @@ TEST_CASE("ClGreaterSimpleEndToEndUint8Test") 0, 0, 0, 0, 0, 0, 0, 0 }); ComparisonSimpleEndToEnd<armnn::DataType::QAsymmU8>(clDefaultBackends, - ComparisonOperation::Greater, - expectedOutput); + ComparisonOperation::Greater, + expectedOutput); } TEST_CASE("ClGreaterBroadcastEndToEndTest") @@ -265,8 +266,8 @@ TEST_CASE("ClGreaterBroadcastEndToEndUint8Test") 1, 1, 1, 1, 1, 1 }); ComparisonBroadcastEndToEnd<armnn::DataType::QAsymmU8>(clDefaultBackends, - ComparisonOperation::Greater, - expectedOutput); + ComparisonOperation::Greater, + expectedOutput); } // HardSwish @@ -316,6 +317,49 @@ TEST_CASE("ClInstanceNormalizationNchwEndToEndTest2") InstanceNormalizationNchwEndToEndTest2(clDefaultBackends); } +// Pooling 2D +// Average Pool 2D +TEST_CASE("ClAvgPool2DEndtoEndTestFloat32") +{ + AvgPool2dEndToEnd<DataType::Float32>(clDefaultBackends); +} + +TEST_CASE("ClAvgPool2DEndtoEndTestFloat16") +{ + AvgPool2dEndToEndFloat16<DataType::Float16>(clDefaultBackends); +} + +TEST_CASE("ClAvgPool2DIgnoreValueEndtoEndTestFloat32") +{ + AvgPool2dEndToEnd<DataType::Float32>(clDefaultBackends, PaddingMethod::IgnoreValue); +} + +// Max Pool 2D +TEST_CASE("ClMaxPool2DEndtoEndTestFloat32") +{ + MaxPool2dEndToEnd<DataType::Float32>(clDefaultBackends); +} + +TEST_CASE("ClMaxPool2DEndtoEndTestFloat16") +{ + MaxPool2dEndToEndFloat16<DataType::Float16>(clDefaultBackends); +} + +TEST_CASE("ClMaxPool2DIgnoreValueEndtoEndTestFloat32") +{ + MaxPool2dEndToEnd<DataType::Float32>(clDefaultBackends, PaddingMethod::IgnoreValue); +} + +TEST_CASE("ClMaxPool2DTwoLayerEndtoEndTestFloat32") +{ + MaxPool2dTwoLayerEndToEnd<DataType::Float32>(clDefaultBackends); +} + +TEST_CASE("ClMaxPool2DThreeLayerEndtoEndTestFloat32") +{ + MaxPool2dThreeLayerEndToEnd<DataType::Float32>(clDefaultBackends); +} + // Fill TEST_CASE("ClFillEndToEndTest") { |