aboutsummaryrefslogtreecommitdiff
path: root/delegate/src/test/TransposeTest.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'delegate/src/test/TransposeTest.cpp')
-rw-r--r--delegate/src/test/TransposeTest.cpp46
1 files changed, 46 insertions, 0 deletions
diff --git a/delegate/src/test/TransposeTest.cpp b/delegate/src/test/TransposeTest.cpp
new file mode 100644
index 0000000000..67751e325a
--- /dev/null
+++ b/delegate/src/test/TransposeTest.cpp
@@ -0,0 +1,46 @@
+//
+// Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#include "TransposeTestHelper.hpp"
+
+#include <armnn_delegate.hpp>
+
+#include <doctest/doctest.h>
+#include <flatbuffers/flatbuffers.h>
+
+namespace armnnDelegate
+{
+
+TEST_SUITE ("Transpose_GpuAccTests")
+{
+
+TEST_CASE ("Transpose_Float32_GpuAcc_Test")
+{
+ std::vector<armnn::BackendId> backends = {armnn::Compute::GpuAcc};
+ TransposeFP32Test(backends);
+}
+
+}
+
+TEST_SUITE ("Transpose_CpuAccTests")
+{
+
+TEST_CASE ("Transpose_Float32_CpuAcc_Test")
+{
+ std::vector<armnn::BackendId> backends = {armnn::Compute::CpuAcc};
+ TransposeFP32Test(backends);
+}
+
+}
+
+TEST_SUITE ("Transpose_CpuRefTests")
+{
+TEST_CASE ("Transpose_Float32_CpuRef_Test")
+{
+ std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef };
+ TransposeFP32Test(backends);
+}
+}
+} // namespace armnnDelegate