diff options
author | James Ward <james.ward@arm.com> | 2020-11-09 11:57:47 +0000 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2020-11-16 11:31:54 +0000 |
commit | f89964ec2f5e66a0c87ca4cc1535f616a7c38afa (patch) | |
tree | e1bde82326fccebcb5765de706f0524c1bf4be01 /delegate/src/test/TransposeTest.cpp | |
parent | 32ca144fc8b4f0a1e2eda274da55ffd0a6016c02 (diff) | |
download | armnn-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.cpp | 46 |
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 |