aboutsummaryrefslogtreecommitdiff
path: root/delegate/test/PreluTest.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'delegate/test/PreluTest.cpp')
-rw-r--r--delegate/test/PreluTest.cpp96
1 files changed, 12 insertions, 84 deletions
diff --git a/delegate/test/PreluTest.cpp b/delegate/test/PreluTest.cpp
index 3240eafe2b..990d91a33f 100644
--- a/delegate/test/PreluTest.cpp
+++ b/delegate/test/PreluTest.cpp
@@ -1,24 +1,15 @@
//
-// 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 "PreluTestHelper.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 {
-void PreluFloatSimpleTest(std::vector <armnn::BackendId>& backends, bool isAlphaConst, bool isDynamicOutput = false)
+void PreluFloatSimpleTest(bool isAlphaConst, bool isDynamicOutput = false)
{
std::vector<int32_t> inputShape { 1, 2, 3 };
std::vector<int32_t> alphaShape { 1 };
@@ -35,7 +26,6 @@ void PreluFloatSimpleTest(std::vector <armnn::BackendId>& backends, bool isAlpha
PreluTest(tflite::BuiltinOperator_PRELU,
::tflite::TensorType_FLOAT32,
- backends,
inputShape,
alphaShape,
outputShape,
@@ -45,91 +35,29 @@ void PreluFloatSimpleTest(std::vector <armnn::BackendId>& backends, bool isAlpha
isAlphaConst);
}
-TEST_SUITE("Prelu_CpuRefTests")
-{
-
-TEST_CASE ("PreluFp32SimpleConstTest_CpuRef_Test")
-{
- std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef };
- PreluFloatSimpleTest(backends, true);
-}
-
-TEST_CASE ("PreluFp32SimpleTest_CpuRef_Test")
-{
- std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef };
- PreluFloatSimpleTest(backends, false);
-}
-
-TEST_CASE ("PreluFp32SimpleConstDynamicTest_CpuRef_Test")
-{
- std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef };
- PreluFloatSimpleTest(backends, true, true);
-}
-
-TEST_CASE ("PreluFp32SimpleDynamicTest_CpuRef_Test")
-{
- std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef };
- PreluFloatSimpleTest(backends, false, true);
-}
-
-} // TEST_SUITE("Prelu_CpuRefTests")
-
-TEST_SUITE("Prelu_CpuAccTests")
-{
-
-TEST_CASE ("PreluFp32SimpleConstTest_CpuAcc_Test")
-{
- std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc };
- PreluFloatSimpleTest(backends, true);
-}
-
-TEST_CASE ("PreluFp32SimpleTest_CpuAcc_Test")
-{
- std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc };
- PreluFloatSimpleTest(backends, false);
-}
-
-TEST_CASE ("PreluFp32SimpleConstDynamicTest_CpuAcc_Test")
-{
- std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc };
- PreluFloatSimpleTest(backends, true, true);
-}
-
-TEST_CASE ("PreluFp32SimpleDynamicTest_CpuAcc_Test")
-{
- std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc };
- PreluFloatSimpleTest(backends, false, true);
-}
-
-} // TEST_SUITE("Prelu_CpuAccTests")
-
-TEST_SUITE("Prelu_GpuAccTests")
+TEST_SUITE("PreluTests")
{
-TEST_CASE ("PreluFp32SimpleConstTest_GpuAcc_Test")
+TEST_CASE ("PreluFp32SimpleConstTest_Test")
{
- std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc };
- PreluFloatSimpleTest(backends, true);
+ PreluFloatSimpleTest(true);
}
-TEST_CASE ("PreluFp32SimpleTest_GpuAcc_Test")
+TEST_CASE ("PreluFp32SimpleTest_Test")
{
- std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc };
- PreluFloatSimpleTest(backends, false);
+ PreluFloatSimpleTest(false);
}
-TEST_CASE ("PreluFp32SimpleConstDynamicTest_GpuAcc_Test")
+TEST_CASE ("PreluFp32SimpleConstDynamicTest_Test")
{
- std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc };
- PreluFloatSimpleTest(backends, true, true);
+ PreluFloatSimpleTest(true, true);
}
-TEST_CASE ("PreluFp32SimpleDynamicTest_GpuAcc_Test")
+TEST_CASE ("PreluFp32SimpleDynamicTest_Test")
{
- std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc };
- PreluFloatSimpleTest(backends, false, true);
+ PreluFloatSimpleTest(false, true);
}
-} // TEST_SUITE("Prelu_GpuAccTests")
+} // TEST_SUITE("PreluTests")
} \ No newline at end of file