aboutsummaryrefslogtreecommitdiff
path: root/delegate/test/PackTest.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'delegate/test/PackTest.cpp')
-rw-r--r--delegate/test/PackTest.cpp197
1 files changed, 30 insertions, 167 deletions
diff --git a/delegate/test/PackTest.cpp b/delegate/test/PackTest.cpp
index 51f545ce5e..4f826d9175 100644
--- a/delegate/test/PackTest.cpp
+++ b/delegate/test/PackTest.cpp
@@ -1,21 +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 "PackTestHelper.hpp"
-#include <armnn_delegate.hpp>
-
-#include <flatbuffers/flatbuffers.h>
-
#include <doctest/doctest.h>
namespace armnnDelegate
{
template <typename T>
-void PackFp32Axis0Test(tflite::TensorType tensorType, std::vector<armnn::BackendId>& backends)
+void PackFp32Axis0Test(tflite::TensorType tensorType)
{
std::vector<int32_t> inputShape { 3, 2, 3 };
std::vector<int32_t> expectedOutputShape { 2, 3, 2, 3 };
@@ -69,16 +65,16 @@ void PackFp32Axis0Test(tflite::TensorType tensorType, std::vector<armnn::Backend
PackTest<T>(tflite::BuiltinOperator_PACK,
tensorType,
- backends,
inputShape,
expectedOutputShape,
inputValues,
expectedOutputValues,
+ {},
0);
}
template <typename T>
-void PackFp32Axis1Test(tflite::TensorType tensorType, std::vector<armnn::BackendId>& backends)
+void PackFp32Axis1Test(tflite::TensorType tensorType)
{
std::vector<int32_t> inputShape { 3, 2, 3 };
std::vector<int32_t> expectedOutputShape { 3, 2, 2, 3 };
@@ -133,16 +129,16 @@ void PackFp32Axis1Test(tflite::TensorType tensorType, std::vector<armnn::Backend
PackTest<T>(tflite::BuiltinOperator_PACK,
tensorType,
- backends,
inputShape,
expectedOutputShape,
inputValues,
expectedOutputValues,
+ {},
1);
}
template <typename T>
-void PackFp32Axis2Test(tflite::TensorType tensorType, std::vector<armnn::BackendId>& backends)
+void PackFp32Axis2Test(tflite::TensorType tensorType)
{
std::vector<int32_t> inputShape { 3, 2, 3 };
std::vector<int32_t> expectedOutputShape { 3, 2, 2, 3 };
@@ -195,16 +191,16 @@ void PackFp32Axis2Test(tflite::TensorType tensorType, std::vector<armnn::Backend
PackTest<T>(tflite::BuiltinOperator_PACK,
tensorType,
- backends,
inputShape,
expectedOutputShape,
inputValues,
expectedOutputValues,
+ {},
2);
}
template <typename T>
-void PackFp32Axis3Test(tflite::TensorType tensorType, std::vector<armnn::BackendId>& backends)
+void PackFp32Axis3Test(tflite::TensorType tensorType)
{
std::vector<int32_t> inputShape { 3, 2, 3 };
std::vector<int32_t> expectedOutputShape { 3, 2, 3, 2 };
@@ -265,16 +261,16 @@ void PackFp32Axis3Test(tflite::TensorType tensorType, std::vector<armnn::Backend
PackTest<T>(tflite::BuiltinOperator_PACK,
tflite::TensorType_FLOAT32,
- backends,
inputShape,
expectedOutputShape,
inputValues,
expectedOutputValues,
+ {},
3);
}
template <typename T>
-void PackFp32Inputs3Test(tflite::TensorType tensorType, std::vector<armnn::BackendId>& backends)
+void PackFp32Inputs3Test(tflite::TensorType tensorType)
{
std::vector<int32_t> inputShape { 3, 3 };
std::vector<int32_t> expectedOutputShape { 3, 3, 3 };
@@ -318,196 +314,63 @@ void PackFp32Inputs3Test(tflite::TensorType tensorType, std::vector<armnn::Backe
PackTest<T>(tflite::BuiltinOperator_PACK,
tensorType,
- backends,
inputShape,
expectedOutputShape,
inputValues,
expectedOutputValues,
+ {},
1);
}
-TEST_SUITE("Pack_CpuAccTests")
-{
-
-// Fp32
-TEST_CASE ("Pack_Fp32_Axis0_CpuAcc_Test")
-{
- std::vector<armnn::BackendId> backends = {armnn::Compute::CpuAcc};
- PackFp32Axis0Test<float>(tflite::TensorType_FLOAT32, backends);
-}
-
-TEST_CASE ("Pack_Fp32_Axis1_CpuAcc_Test")
-{
- std::vector<armnn::BackendId> backends = {armnn::Compute::CpuAcc};
- PackFp32Axis1Test<float>(tflite::TensorType_FLOAT32, backends);
-}
-
-TEST_CASE ("Pack_Fp32_Axis2_CpuAcc_Test")
-{
- std::vector<armnn::BackendId> backends = {armnn::Compute::CpuAcc};
- PackFp32Axis2Test<float>(tflite::TensorType_FLOAT32, backends);
-}
-
-TEST_CASE ("Pack_Fp32_Axis3_CpuAcc_Test")
-{
- std::vector<armnn::BackendId> backends = {armnn::Compute::CpuAcc};
- PackFp32Axis3Test<float>(tflite::TensorType_FLOAT32, backends);
-}
-
-TEST_CASE ("Pack_Fp32_Inputs3_CpuAcc_Test")
-{
- std::vector<armnn::BackendId> backends = {armnn::Compute::CpuAcc};
- PackFp32Inputs3Test<float>(tflite::TensorType_FLOAT32, backends);
-}
-
-// Uint8
-TEST_CASE ("Pack_Uint8_Axis0_CpuAcc_Test")
-{
- std::vector<armnn::BackendId> backends = {armnn::Compute::CpuAcc};
- PackFp32Axis0Test<uint8_t>(tflite::TensorType_UINT8, backends);
-}
-
-TEST_CASE ("Pack_Uint8_Inputs3_CpuAcc_Test")
-{
- std::vector<armnn::BackendId> backends = {armnn::Compute::CpuAcc};
- PackFp32Inputs3Test<uint8_t>(tflite::TensorType_UINT8, backends);
-}
-
-// Uint8
-TEST_CASE ("Pack_Int8_Axis0_CpuAcc_Test")
-{
- std::vector<armnn::BackendId> backends = {armnn::Compute::CpuAcc};
- PackFp32Axis0Test<int8_t>(tflite::TensorType_INT8, backends);
-}
-
-TEST_CASE ("Pack_Int8_Inputs3_CpuAcc_Test")
-{
- std::vector<armnn::BackendId> backends = {armnn::Compute::CpuAcc};
- PackFp32Inputs3Test<int8_t>(tflite::TensorType_INT8, backends);
-}
-
-}
-
-TEST_SUITE("Pack_GpuAccTests")
-{
-
-// Fp32
-TEST_CASE ("Pack_Fp32_Axis0_GpuAcc_Test")
-{
- std::vector<armnn::BackendId> backends = {armnn::Compute::GpuAcc};
- PackFp32Axis0Test<float>(tflite::TensorType_FLOAT32, backends);
-}
-
-TEST_CASE ("Pack_Fp32_Axis1_GpuAcc_Test")
-{
- std::vector<armnn::BackendId> backends = {armnn::Compute::GpuAcc};
- PackFp32Axis1Test<float>(tflite::TensorType_FLOAT32, backends);
-}
-
-TEST_CASE ("Pack_Fp32_Axis2_GpuAcc_Test")
-{
- std::vector<armnn::BackendId> backends = {armnn::Compute::GpuAcc};
- PackFp32Axis2Test<float>(tflite::TensorType_FLOAT32, backends);
-}
-
-TEST_CASE ("Pack_Fp32_Axis3_GpuAcc_Test")
-{
- std::vector<armnn::BackendId> backends = {armnn::Compute::GpuAcc};
- PackFp32Axis3Test<float>(tflite::TensorType_FLOAT32, backends);
-}
-
-TEST_CASE ("Pack_Fp32_Inputs3_GpuAcc_Test")
-{
- std::vector<armnn::BackendId> backends = {armnn::Compute::GpuAcc};
- PackFp32Inputs3Test<float>(tflite::TensorType_FLOAT32, backends);
-}
-
-// Uint8
-TEST_CASE ("Pack_Uint8_Axis0_GpuAcc_Test")
-{
- std::vector<armnn::BackendId> backends = {armnn::Compute::GpuAcc};
- PackFp32Axis0Test<uint8_t>(tflite::TensorType_UINT8, backends);
-}
-
-TEST_CASE ("Pack_Uint8_Inputs3_GpuAcc_Test")
-{
- std::vector<armnn::BackendId> backends = {armnn::Compute::GpuAcc};
- PackFp32Inputs3Test<uint8_t>(tflite::TensorType_UINT8, backends);
-}
-
-// Int8
-TEST_CASE ("Pack_Int8_Axis0_GpuAcc_Test")
-{
- std::vector<armnn::BackendId> backends = {armnn::Compute::GpuAcc};
- PackFp32Axis0Test<int8_t>(tflite::TensorType_INT8, backends);
-}
-
-TEST_CASE ("Pack_Int8_Inputs3_GpuAcc_Test")
-{
- std::vector<armnn::BackendId> backends = {armnn::Compute::GpuAcc};
- PackFp32Inputs3Test<int8_t>(tflite::TensorType_INT8, backends);
-}
-
-}
-
-TEST_SUITE("Pack_CpuRefTests")
+TEST_SUITE("PackTests")
{
// Fp32
-TEST_CASE ("Pack_Fp32_Axis0_CpuRef_Test")
+TEST_CASE ("Pack_Fp32_Axis0_Test")
{
- std::vector<armnn::BackendId> backends = {armnn::Compute::CpuRef};
- PackFp32Axis0Test<float>(tflite::TensorType_FLOAT32, backends);
+ PackFp32Axis0Test<float>(tflite::TensorType_FLOAT32);
}
-TEST_CASE ("Pack_Fp32_Axis1_CpuRef_Test")
+TEST_CASE ("Pack_Fp32_Axis1_Test")
{
- std::vector<armnn::BackendId> backends = {armnn::Compute::CpuRef};
- PackFp32Axis1Test<float>(tflite::TensorType_FLOAT32, backends);
+ PackFp32Axis1Test<float>(tflite::TensorType_FLOAT32);
}
-TEST_CASE ("Pack_Fp32_Axis2_CpuRef_Test")
+TEST_CASE ("Pack_Fp32_Axis2_Test")
{
- std::vector<armnn::BackendId> backends = {armnn::Compute::CpuRef};
- PackFp32Axis2Test<float>(tflite::TensorType_FLOAT32, backends);
+ PackFp32Axis2Test<float>(tflite::TensorType_FLOAT32);
}
-TEST_CASE ("Pack_Fp32_Axis3_CpuRef_Test")
+TEST_CASE ("Pack_Fp32_Axis3_Test")
{
- std::vector<armnn::BackendId> backends = {armnn::Compute::CpuRef};
- PackFp32Axis3Test<float>(tflite::TensorType_FLOAT32, backends);
+ PackFp32Axis3Test<float>(tflite::TensorType_FLOAT32);
}
-TEST_CASE ("Pack_Fp32_Inputs3_CpuRef_Test")
+TEST_CASE ("Pack_Fp32_Inputs3_Test")
{
- std::vector<armnn::BackendId> backends = {armnn::Compute::CpuRef};
- PackFp32Inputs3Test<float>(tflite::TensorType_FLOAT32, backends);
+ PackFp32Inputs3Test<float>(tflite::TensorType_FLOAT32);
}
// Uint8
-TEST_CASE ("Pack_Uint8_Axis0_CpuRef_Test")
+TEST_CASE ("Pack_Uint8_Axis0_Test")
{
- std::vector<armnn::BackendId> backends = {armnn::Compute::CpuRef};
- PackFp32Axis0Test<uint8_t>(tflite::TensorType_UINT8, backends);
+ PackFp32Axis0Test<uint8_t>(tflite::TensorType_UINT8);
}
-TEST_CASE ("Pack_Uint8_Inputs3_CpuRef_Test")
+TEST_CASE ("Pack_Uint8_Inputs3_Test")
{
- std::vector<armnn::BackendId> backends = {armnn::Compute::CpuRef};
- PackFp32Inputs3Test<uint8_t>(tflite::TensorType_UINT8, backends);
+ PackFp32Inputs3Test<uint8_t>(tflite::TensorType_UINT8);
}
// Int8
-TEST_CASE ("Pack_Int8_Axis0_CpuRef_Test")
+TEST_CASE ("Pack_Int8_Axis0_Test")
{
- std::vector<armnn::BackendId> backends = {armnn::Compute::CpuRef};
- PackFp32Axis0Test<int8_t>(tflite::TensorType_INT8, backends);
+ PackFp32Axis0Test<int8_t>(tflite::TensorType_INT8);
}
-TEST_CASE ("Pack_Int8_Inputs3_CpuRef_Test")
+TEST_CASE ("Pack_Int8_Inputs3_Test")
{
- std::vector<armnn::BackendId> backends = {armnn::Compute::CpuRef};
- PackFp32Inputs3Test<int8_t>(tflite::TensorType_INT8, backends);
+ PackFp32Inputs3Test<int8_t>(tflite::TensorType_INT8);
}
}
generated by cgit v1.2.1 (git 2.23.0) at 2024-06-01 11:06:50 +0000