diff options
author | Matthew Sloyan <matthew.sloyan@arm.com> | 2023-03-30 10:12:08 +0100 |
---|---|---|
committer | ryan.oshea3 <ryan.oshea3@arm.com> | 2023-04-05 20:36:32 +0000 |
commit | ebe392df1635790bf21714549adb97f2f75559e1 (patch) | |
tree | 6fb8e56cc755d7c47a62bbe72c54b6ca5445377d /delegate/test/TransposeTest.cpp | |
parent | ac9607f401dc30003aa97bd179a06d6b8a32139f (diff) | |
download | armnn-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.cpp | 29 |
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 |