aboutsummaryrefslogtreecommitdiff
path: root/delegate/test/TransposeTest.cpp
diff options
context:
space:
mode:
authorMatthew Sloyan <matthew.sloyan@arm.com>2023-03-30 10:12:08 +0100
committerryan.oshea3 <ryan.oshea3@arm.com>2023-04-05 20:36:32 +0000
commitebe392df1635790bf21714549adb97f2f75559e1 (patch)
tree6fb8e56cc755d7c47a62bbe72c54b6ca5445377d /delegate/test/TransposeTest.cpp
parentac9607f401dc30003aa97bd179a06d6b8a32139f (diff)
downloadarmnn-ebe392df1635790bf21714549adb97f2f75559e1.tar.gz
IVGCVSW-7562 Implement DelegateTestInterpreter for classic delegate
* Updated all tests to use new DelegateTestInterpreter. * Fixed some unit tests where the shape was incorrect. * Add file identifier to FlatBuffersBuilder, as it is required for validation when creating the model using new API. Signed-off-by: Matthew Sloyan <matthew.sloyan@arm.com> Change-Id: I1c4f5464367b35d4528571fa94d14bfaef18fb4d
Diffstat (limited to 'delegate/test/TransposeTest.cpp')
-rw-r--r--delegate/test/TransposeTest.cpp29
1 files changed, 27 insertions, 2 deletions
diff --git a/delegate/test/TransposeTest.cpp b/delegate/test/TransposeTest.cpp
index c210128ac8..cb3b327b13 100644
--- a/delegate/test/TransposeTest.cpp
+++ b/delegate/test/TransposeTest.cpp
@@ -13,6 +13,28 @@
namespace armnnDelegate
{
+void TransposeFP32Test(std::vector<armnn::BackendId>& backends)
+{
+ // set test input data
+ std::vector<int32_t> input0Shape {4, 2, 3};
+ std::vector<int32_t> inputPermVecShape {3};
+ std::vector<int32_t> outputShape {3, 4, 2};
+
+ std::vector<float> input0Values = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11,
+ 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23};
+ std::vector<int32_t> inputPermVec = {2, 0, 1};
+ std::vector<float> expectedOutputValues = {0, 3, 6, 9, 12, 15, 18, 21, 1, 4, 7, 10,
+ 13, 16, 19, 22, 2, 5, 8, 11, 14, 17, 20, 23};
+
+ TransposeTest<float>(backends,
+ input0Shape,
+ inputPermVecShape,
+ outputShape,
+ input0Values,
+ inputPermVec,
+ expectedOutputValues);
+}
+
TEST_SUITE ("Transpose_GpuAccTests")
{
@@ -37,10 +59,13 @@ TEST_CASE ("Transpose_Float32_CpuAcc_Test")
TEST_SUITE ("Transpose_CpuRefTests")
{
+
TEST_CASE ("Transpose_Float32_CpuRef_Test")
{
- std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef };
- TransposeFP32Test(backends);
+ std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef };
+ TransposeFP32Test(backends);
}
+
}
+
} // namespace armnnDelegate