diff options
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 |