diff options
Diffstat (limited to 'delegate/test/SpaceDepthTest.cpp')
| -rw-r--r-- | delegate/test/SpaceDepthTest.cpp | 118 |
1 files changed, 21 insertions, 97 deletions
diff --git a/delegate/test/SpaceDepthTest.cpp b/delegate/test/SpaceDepthTest.cpp index 6cffba33eb..add51b4a21 100644 --- a/delegate/test/SpaceDepthTest.cpp +++ b/delegate/test/SpaceDepthTest.cpp @@ -1,20 +1,16 @@ // -// 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 "SpaceDepthTestHelper.hpp" -#include <armnn_delegate.hpp> - -#include <flatbuffers/flatbuffers.h> - #include <doctest/doctest.h> namespace armnnDelegate { -void DepthToSpaceFp32Test(std::vector<armnn::BackendId>& backends, int blockSize) +void DepthToSpaceFp32Test(int blockSize) { // Set input data std::vector<int32_t> inputShape { 1, 2, 2, 4 }; @@ -32,15 +28,15 @@ void DepthToSpaceFp32Test(std::vector<armnn::BackendId>& backends, int blockSize SpaceDepthTest<float>(tflite::BuiltinOperator_DEPTH_TO_SPACE, ::tflite::TensorType_FLOAT32, - backends, inputShape, outputShape, inputValues, expectedOutputValues, + {}, blockSize); } -void DepthToSpaceUint8Test(std::vector<armnn::BackendId>& backends, int blockSize) +void DepthToSpaceUint8Test(int blockSize) { // Set input data std::vector<int32_t> inputShape { 2, 1, 1, 4 }; @@ -54,15 +50,15 @@ void DepthToSpaceUint8Test(std::vector<armnn::BackendId>& backends, int blockSiz SpaceDepthTest<uint8_t>(tflite::BuiltinOperator_DEPTH_TO_SPACE, ::tflite::TensorType_UINT8, - backends, inputShape, outputShape, inputValues, expectedOutputValues, + {}, blockSize); } -void SpaceToDepthFp32Test(std::vector<armnn::BackendId>& backends, int blockSize) +void SpaceToDepthFp32Test(int blockSize) { // Set input data std::vector<int32_t> inputShape { 1, 2, 2, 2 }; @@ -73,15 +69,15 @@ void SpaceToDepthFp32Test(std::vector<armnn::BackendId>& backends, int blockSize SpaceDepthTest<float>(tflite::BuiltinOperator_SPACE_TO_DEPTH, ::tflite::TensorType_FLOAT32, - backends, inputShape, outputShape, inputValues, expectedOutputValues, + {}, blockSize); } -void SpaceToDepthUint8Test(std::vector<armnn::BackendId>& backends, int blockSize) +void SpaceToDepthUint8Test(int blockSize) { // Set input data std::vector<int32_t> inputShape { 1, 2, 2, 1 }; @@ -92,115 +88,43 @@ void SpaceToDepthUint8Test(std::vector<armnn::BackendId>& backends, int blockSiz SpaceDepthTest<uint8_t>(tflite::BuiltinOperator_SPACE_TO_DEPTH, ::tflite::TensorType_UINT8, - backends, inputShape, outputShape, inputValues, expectedOutputValues, + {}, blockSize); } -TEST_SUITE("DepthToSpace_CpuRefTests") -{ - -TEST_CASE ("DepthToSpaceFp32Test_CpuRef_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef }; - DepthToSpaceFp32Test(backends, 2); -} - -TEST_CASE ("DepthToSpaceUint8Test_CpuRef_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef }; - DepthToSpaceUint8Test(backends, 2); -} - -} // TEST_SUITE("DepthToSpace_CpuRefTests") - - -TEST_SUITE("DepthToSpace_CpuAccTests") -{ - -TEST_CASE ("DepthToSpaceFp32Test_CpuAcc_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc }; - DepthToSpaceFp32Test(backends, 2); -} - -TEST_CASE ("DepthToSpaceUint8Test_CpuAcc_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc }; - DepthToSpaceUint8Test(backends, 2); -} - -} // TEST_SUITE("DepthToSpace_CpuAccTests") - -TEST_SUITE("DepthToSpace_GpuAccTests") +TEST_SUITE("DepthToSpaceTests") { -TEST_CASE ("DepthToSpaceFp32Test_GpuAcc_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc }; - DepthToSpaceFp32Test(backends, 2); -} - -TEST_CASE ("DepthToSpaceUint8Test_GpuAcc_Test") +TEST_CASE ("DepthToSpaceFp32Test_Test") { - std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc }; - DepthToSpaceUint8Test(backends, 2); + DepthToSpaceFp32Test(2); } -} // TEST_SUITE("DepthToSpace_GpuAccTests") - -TEST_SUITE("SpaceToDepth_CpuRefTests") -{ - -TEST_CASE ("SpaceToDepthFp32Test_CpuRef_Test") +TEST_CASE ("DepthToSpaceUint8Test_Test") { - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef }; - SpaceToDepthFp32Test(backends, 2); + DepthToSpaceUint8Test(2); } -TEST_CASE ("SpaceToDepthUint8Test_CpuRef_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef }; - SpaceToDepthUint8Test(backends, 2); -} - -} // TEST_SUITE("SpaceToDepth_CpuRefTests") - -TEST_SUITE("SpaceToDepth_CpuAccTests") -{ - -TEST_CASE ("SpaceToDepthFp32Test_CpuAcc_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc }; - SpaceToDepthFp32Test(backends, 2); -} - -TEST_CASE ("SpaceToDepthUint8Test_CpuAcc_Test") -{ - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc }; - SpaceToDepthUint8Test(backends, 2); -} +} // TEST_SUITE("DepthToSpaceTests") -} // TEST_SUITE("SpaceToDepth_CpuAccTests") -TEST_SUITE("SpaceToDepth_GpuAccTests") +TEST_SUITE("SpaceToDepthTests") { -TEST_CASE ("SpaceToDepthFp32Test_GpuAcc_Test") +TEST_CASE ("SpaceToDepthFp32Test_Test") { - std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc }; - SpaceToDepthFp32Test(backends, 2); + SpaceToDepthFp32Test(2); } -TEST_CASE ("SpaceToDepthUint8Test_GpuAcc_Test") +TEST_CASE ("SpaceToDepthUint8Test_Test") { - std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc }; - SpaceToDepthUint8Test(backends, 2); + SpaceToDepthUint8Test(2); } -} // TEST_SUITE("SpaceToDepth_GpuAccTests") +} // TEST_SUITE("SpaceToDepthTests") } // namespace armnnDelegate |