aboutsummaryrefslogtreecommitdiff
path: root/delegate/src/test/TransposeTest.cpp
diff options
context:
space:
mode:
authorJames Ward <james.ward@arm.com>2020-11-09 11:57:47 +0000
committerJim Flynn <jim.flynn@arm.com>2020-11-16 11:31:54 +0000
commitf89964ec2f5e66a0c87ca4cc1535f616a7c38afa (patch)
treee1bde82326fccebcb5765de706f0524c1bf4be01 /delegate/src/test/TransposeTest.cpp
parent32ca144fc8b4f0a1e2eda274da55ffd0a6016c02 (diff)
downloadarmnn-f89964ec2f5e66a0c87ca4cc1535f616a7c38afa.tar.gz
IVGCVSW-5385 TfLiteDelegate: Implement the Transpose operator
Signed-off-by: James Ward <james.ward@arm.com> Change-Id: Iea3d7ecccb82d85ec2d2c5cfdcdaf692236a60aa
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