diff options
Diffstat (limited to 'delegate/test/FillTest.cpp')
| -rw-r--r-- | delegate/test/FillTest.cpp | 130 |
1 files changed, 21 insertions, 109 deletions
diff --git a/delegate/test/FillTest.cpp b/delegate/test/FillTest.cpp index e9def8de41..5cde6dd8eb 100644 --- a/delegate/test/FillTest.cpp +++ b/delegate/test/FillTest.cpp @@ -1,22 +1,17 @@ // -// Copyright © 2021, 2023 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2021, 2023-2024 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #include "FillTestHelper.hpp" -#include <armnn_delegate.hpp> - -#include <flatbuffers/flatbuffers.h> - #include <doctest/doctest.h> namespace armnnDelegate { -void Fill2dTest(std::vector<armnn::BackendId>& backends, - tflite::BuiltinOperator fillOperatorCode = tflite::BuiltinOperator_FILL, - float fill = 2.0f ) +void Fill2dTest(tflite::BuiltinOperator fillOperatorCode = tflite::BuiltinOperator_FILL, + float fill = 2.0f ) { std::vector<int32_t> inputShape { 2 }; std::vector<int32_t> tensorShape { 2, 2 }; @@ -25,16 +20,14 @@ void Fill2dTest(std::vector<armnn::BackendId>& backends, FillTest<float>(fillOperatorCode, ::tflite::TensorType_FLOAT32, - backends, inputShape, tensorShape, expectedOutputValues, fill); } -void Fill3dTest(std::vector<armnn::BackendId>& backends, - tflite::BuiltinOperator fillOperatorCode = tflite::BuiltinOperator_FILL, - float fill = 5.0f ) +void Fill3dTest(tflite::BuiltinOperator fillOperatorCode = tflite::BuiltinOperator_FILL, + float fill = 5.0f ) { std::vector<int32_t> inputShape { 3 }; std::vector<int32_t> tensorShape { 3, 3, 3 }; @@ -52,16 +45,14 @@ void Fill3dTest(std::vector<armnn::BackendId>& backends, FillTest<float>(fillOperatorCode, ::tflite::TensorType_FLOAT32, - backends, inputShape, tensorShape, expectedOutputValues, fill); } -void Fill4dTest(std::vector<armnn::BackendId>& backends, - tflite::BuiltinOperator fillOperatorCode = tflite::BuiltinOperator_FILL, - float fill = 3.0f ) +void Fill4dTest(tflite::BuiltinOperator fillOperatorCode = tflite::BuiltinOperator_FILL, + float fill = 3.0f ) { std::vector<int32_t> inputShape { 4 }; std::vector<int32_t> tensorShape { 2, 2, 4, 4 }; @@ -87,16 +78,14 @@ void Fill4dTest(std::vector<armnn::BackendId>& backends, FillTest<float>(fillOperatorCode, ::tflite::TensorType_FLOAT32, - backends, inputShape, tensorShape, expectedOutputValues, fill); } -void FillInt32Test(std::vector<armnn::BackendId>& backends, - tflite::BuiltinOperator fillOperatorCode = tflite::BuiltinOperator_FILL, - int32_t fill = 2 ) +void FillInt32Test(tflite::BuiltinOperator fillOperatorCode = tflite::BuiltinOperator_FILL, + int32_t fill = 2 ) { std::vector<int32_t> inputShape { 2 }; std::vector<int32_t> tensorShape { 2, 2 }; @@ -105,116 +94,39 @@ void FillInt32Test(std::vector<armnn::BackendId>& backends, FillTest<int32_t>(fillOperatorCode, ::tflite::TensorType_INT32, - backends, inputShape, tensorShape, expectedOutputValues, fill); } -TEST_SUITE("Fill_CpuRefTests") -{ - -TEST_CASE ("Fill2d_CpuRef_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef }; - Fill2dTest(backends); -} - -TEST_CASE ("Fill3d_CpuRef_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef }; - Fill3dTest(backends); -} - -TEST_CASE ("Fill3d_CpuRef_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef }; - Fill3dTest(backends); -} - -TEST_CASE ("Fill4d_CpuRef_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef }; - Fill4dTest(backends); -} - -TEST_CASE ("FillInt32_CpuRef_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef }; - FillInt32Test(backends); -} - -} - -TEST_SUITE("Fill_CpuAccTests") +TEST_SUITE("FillTests") { -TEST_CASE ("Fill2d_CpuAcc_Test") +TEST_CASE ("Fill2d_Test") { - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc }; - Fill2dTest(backends); + Fill2dTest(); } -TEST_CASE ("Fill3d_CpuAcc_Test") +TEST_CASE ("Fill3d_Test") { - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc }; - Fill3dTest(backends); + Fill3dTest(); } -TEST_CASE ("Fill3d_CpuAcc_Test") +TEST_CASE ("Fill3d_Test") { - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc }; - Fill3dTest(backends); + Fill3dTest(); } -TEST_CASE ("Fill4d_CpuAcc_Test") +TEST_CASE ("Fill4d_Test") { - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc }; - Fill4dTest(backends); + Fill4dTest(); } -TEST_CASE ("FillInt32_CpuAcc_Test") +TEST_CASE ("FillInt32_Test") { - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc }; - FillInt32Test(backends); -} - -} - -TEST_SUITE("Fill_GpuAccTests") -{ - -TEST_CASE ("Fill2d_GpuAcc_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc }; - Fill2dTest(backends); -} - -TEST_CASE ("Fill3d_GpuAcc_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc }; - Fill3dTest(backends); -} - -TEST_CASE ("Fill3d_GpuAcc_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc }; - Fill3dTest(backends); -} - -TEST_CASE ("Fill4d_GpuAcc_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc }; - Fill4dTest(backends); -} - -TEST_CASE ("FillInt32_GpuAcc_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc }; - FillInt32Test(backends); -} - + FillInt32Test(); } +} // End of FillTests suite. } // namespace armnnDelegate
\ No newline at end of file |