diff options
Diffstat (limited to 'delegate/test/Pooling3dTest.cpp')
-rw-r--r-- | delegate/test/Pooling3dTest.cpp | 176 |
1 files changed, 33 insertions, 143 deletions
diff --git a/delegate/test/Pooling3dTest.cpp b/delegate/test/Pooling3dTest.cpp index a79ee33b0a..04a31f8ee5 100644 --- a/delegate/test/Pooling3dTest.cpp +++ b/delegate/test/Pooling3dTest.cpp @@ -1,19 +1,10 @@ // -// Copyright © 2022-2023 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2022-2024 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #include "Pooling3dTestHelper.hpp" -#include <armnn_delegate.hpp> - -#include <flatbuffers/flatbuffers.h> -#include <tensorflow/lite/interpreter.h> -#include <tensorflow/lite/kernels/register.h> -#include <tensorflow/lite/model.h> - -#include <tensorflow/lite/version.h> - #include <doctest/doctest.h> namespace armnnDelegate @@ -22,7 +13,7 @@ namespace armnnDelegate // Pool3D custom op was only added in tflite r2.6. #if defined(ARMNN_POST_TFLITE_2_5) -void MaxPool3dFP32PaddingValidTest(std::vector<armnn::BackendId>& backends) +void MaxPool3dFP32PaddingValidTest(const std::vector<armnn::BackendId>& backends = {}) { // Set input and expected output data std::vector<int32_t> inputShape = { 1, 2, 3, 4, 1 }; @@ -41,11 +32,11 @@ void MaxPool3dFP32PaddingValidTest(std::vector<armnn::BackendId>& backends) Pooling3dTest<float>(poolType, ::tflite::TensorType_FLOAT32, - backends, inputShape, outputShape, inputValues, expectedOutputValues, + backends, padding, 1, 1, @@ -55,7 +46,7 @@ void MaxPool3dFP32PaddingValidTest(std::vector<armnn::BackendId>& backends) 2); } -void MaxPool3dFP32PaddingSameTest(std::vector<armnn::BackendId>& backends) +void MaxPool3dFP32PaddingSameTest(const std::vector<armnn::BackendId>& backends = {}) { // Set input data and expected output data std::vector<int32_t> inputShape = { 1, 2, 3, 4, 1 }; @@ -74,11 +65,11 @@ void MaxPool3dFP32PaddingSameTest(std::vector<armnn::BackendId>& backends) Pooling3dTest<float>(poolType, ::tflite::TensorType_FLOAT32, - backends, inputShape, outputShape, inputValues, expectedOutputValues, + backends, padding, 1, 1, @@ -88,7 +79,7 @@ void MaxPool3dFP32PaddingSameTest(std::vector<armnn::BackendId>& backends) 2); } -void MaxPool3dFP32H1Test(std::vector<armnn::BackendId>& backends) +void MaxPool3dFP32H1Test(const std::vector<armnn::BackendId>& backends = {}) { // Set input data and expected output data std::vector<int32_t> inputShape = { 1, 2, 3, 4, 1 }; @@ -107,11 +98,11 @@ void MaxPool3dFP32H1Test(std::vector<armnn::BackendId>& backends) Pooling3dTest<float>(poolType, ::tflite::TensorType_FLOAT32, - backends, inputShape, outputShape, inputValues, expectedOutputValues, + backends, padding, 1, 1, @@ -121,7 +112,7 @@ void MaxPool3dFP32H1Test(std::vector<armnn::BackendId>& backends) 2); } -void MaxPool3dFP32Test(std::vector<armnn::BackendId>& backends) +void MaxPool3dFP32Test(const std::vector<armnn::BackendId>& backends = {}) { // Set input data and expected output data std::vector<int32_t> inputShape = { 1, 2, 3, 4, 1 }; @@ -140,11 +131,11 @@ void MaxPool3dFP32Test(std::vector<armnn::BackendId>& backends) Pooling3dTest<float>(poolType, ::tflite::TensorType_FLOAT32, - backends, inputShape, outputShape, inputValues, expectedOutputValues, + backends, padding, 1, 1, @@ -154,7 +145,7 @@ void MaxPool3dFP32Test(std::vector<armnn::BackendId>& backends) 2); } -void AveragePool3dFP32PaddingValidTest(std::vector<armnn::BackendId>& backends) +void AveragePool3dFP32PaddingValidTest(const std::vector<armnn::BackendId>& backends = {}) { // Set input data and expected output data. std::vector<int32_t> inputShape = { 1, 2, 3, 4, 1 }; @@ -173,11 +164,11 @@ void AveragePool3dFP32PaddingValidTest(std::vector<armnn::BackendId>& backends) Pooling3dTest<float>(poolType, ::tflite::TensorType_FLOAT32, - backends, inputShape, outputShape, inputValues, expectedOutputValues, + backends, padding, 1, 1, @@ -187,7 +178,7 @@ void AveragePool3dFP32PaddingValidTest(std::vector<armnn::BackendId>& backends) 2); } -void AveragePool3dFP32PaddingSameTest(std::vector<armnn::BackendId>& backends) +void AveragePool3dFP32PaddingSameTest(const std::vector<armnn::BackendId>& backends = {}) { // Set input data and expected output data std::vector<int32_t> inputShape = { 4, 2, 3, 1, 1 }; @@ -206,11 +197,11 @@ void AveragePool3dFP32PaddingSameTest(std::vector<armnn::BackendId>& backends) Pooling3dTest<float>(poolType, ::tflite::TensorType_FLOAT32, - backends, inputShape, outputShape, inputValues, expectedOutputValues, + backends, padding, 1, 1, @@ -220,7 +211,7 @@ void AveragePool3dFP32PaddingSameTest(std::vector<armnn::BackendId>& backends) 2); } -void AveragePool3dFP32H1Test(std::vector<armnn::BackendId>& backends) +void AveragePool3dFP32H1Test(const std::vector<armnn::BackendId>& backends = {}) { // Set input data and expected output data std::vector<int32_t> inputShape = { 1, 2, 3, 4, 1 }; @@ -239,11 +230,11 @@ void AveragePool3dFP32H1Test(std::vector<armnn::BackendId>& backends) Pooling3dTest<float>(poolType, ::tflite::TensorType_FLOAT32, - backends, inputShape, outputShape, inputValues, expectedOutputValues, + backends, padding, 2, 2, @@ -253,7 +244,7 @@ void AveragePool3dFP32H1Test(std::vector<armnn::BackendId>& backends) 2); } -void AveragePool3dFP32Test(std::vector<armnn::BackendId>& backends) +void AveragePool3dFP32Test(const std::vector<armnn::BackendId>& backends = {}) { // Set input data and expected output data std::vector<int32_t> inputShape = { 4, 3, 2, 1, 1 }; @@ -272,11 +263,11 @@ void AveragePool3dFP32Test(std::vector<armnn::BackendId>& backends) Pooling3dTest<float>(poolType, ::tflite::TensorType_FLOAT32, - backends, inputShape, outputShape, inputValues, expectedOutputValues, + backends, padding, 2, 2, @@ -286,145 +277,44 @@ void AveragePool3dFP32Test(std::vector<armnn::BackendId>& backends) 2); } -TEST_SUITE("Pooling3d_GpuAccTests") -{ - -TEST_CASE ("MaxPooling3d_FP32_GpuAcc_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc }; - MaxPool3dFP32Test(backends); -} - -TEST_CASE ("MaxPooling3d_FP32_PaddingValid_GpuAcc_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc }; - MaxPool3dFP32PaddingValidTest(backends); -} - -TEST_CASE ("MaxPooling3d_FP32_PaddingSame_GpuAcc_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc }; - MaxPool3dFP32PaddingSameTest(backends); -} - -TEST_CASE ("MaxPooling3d_FP32_H1_GpuAcc_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc }; - MaxPool3dFP32H1Test(backends); -} - -TEST_CASE ("AveragePooling3d_FP32_PaddingValid_GpuAcc_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc }; - AveragePool3dFP32PaddingValidTest(backends); -} - -TEST_CASE ("AveragePooling3d_FP32_PaddingSame_GpuAcc_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc }; - AveragePool3dFP32PaddingSameTest(backends); -} - -TEST_CASE ("AveragePooling3d_FP32_H1_GpuAcc_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc }; - AveragePool3dFP32H1Test(backends); -} - -} // TEST_SUITE("Pooling3d_GpuAccTests") - -TEST_SUITE("Pooling3d_CpuAccTests") -{ - -TEST_CASE ("MaxPooling3d_FP32_PaddingValid_CpuAcc_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc }; - MaxPool3dFP32PaddingValidTest(backends); -} - -TEST_CASE ("MaxPooling3d_FP32_PaddingSame_CpuAcc_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc }; - MaxPool3dFP32PaddingSameTest(backends); -} - -TEST_CASE ("MaxPooling3d_FP32_CpuAcc_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc }; - MaxPool3dFP32Test(backends); -} - -TEST_CASE ("MaxPooling3d_FP32_H1_CpuAcc_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc }; - MaxPool3dFP32H1Test(backends); -} - -TEST_CASE ("AveragePooling3d_FP32_PaddingValid_CpuAcc_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc }; - AveragePool3dFP32PaddingValidTest(backends); -} - -TEST_CASE ("AveragePooling3d_FP32_PaddingSame_CpuAcc_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc }; - AveragePool3dFP32PaddingSameTest(backends); -} - -TEST_CASE ("AveragePooling3d_FP32_H1_CpuAcc_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc }; - AveragePool3dFP32H1Test(backends); -} - -} // TEST_SUITE("Pooling3d_CpuAccTests") - -TEST_SUITE("Pooling3d_CpuRefTests") +TEST_SUITE("Pooling3dTests") { -TEST_CASE ("MaxPooling3d_FP32_CpuRef_Test") +TEST_CASE ("MaxPooling3d_FP32_Test") { - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef }; - MaxPool3dFP32Test(backends); + MaxPool3dFP32Test(); } -TEST_CASE ("MaxPooling3d_FP32_PaddingValid_CpuRef_Test") +TEST_CASE ("MaxPooling3d_FP32_PaddingValid_Test") { - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef }; - MaxPool3dFP32PaddingValidTest(backends); + MaxPool3dFP32PaddingValidTest(); } -TEST_CASE ("MaxPooling3d_FP32_PaddingSame_CpuRef_Test") +TEST_CASE ("MaxPooling3d_FP32_PaddingSame_Test") { - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef }; - MaxPool3dFP32PaddingSameTest(backends); + MaxPool3dFP32PaddingSameTest(); } -TEST_CASE ("MaxPooling3d_FP32_H1_CpuRef_Test") +TEST_CASE ("MaxPooling3d_FP32_H1_Test") { - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef }; - MaxPool3dFP32H1Test(backends); + MaxPool3dFP32H1Test(); } -TEST_CASE ("AveragePooling3d_FP32_PaddingValid_CpuRef_Test") +TEST_CASE ("AveragePooling3d_FP32_PaddingValid_Test") { - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef }; - AveragePool3dFP32PaddingValidTest(backends); + AveragePool3dFP32PaddingValidTest(); } -TEST_CASE ("AveragePooling3d_FP32_PaddingSame_CpuRef_Test") +TEST_CASE ("AveragePooling3d_FP32_PaddingSame_Test") { - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef }; - AveragePool3dFP32PaddingSameTest(backends); + AveragePool3dFP32PaddingSameTest(); } -TEST_CASE ("AveragePooling3d_FP32_H1_CpuRef_Test") +TEST_CASE ("AveragePooling3d_FP32_H1_Test") { - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef }; - AveragePool3dFP32H1Test(backends); + AveragePool3dFP32H1Test(); } -} // TEST_SUITE("Pooling3d_CpuRefTests") +} // TEST_SUITE("Pooling3dTests") #endif |