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