diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-09-09 11:22:56 +0100 |
---|---|---|
committer | Kevin May <kevin.may@arm.com> | 2019-09-09 15:18:08 +0000 |
commit | 8fccd86b95be418d1b0397bfd5f48a319f210180 (patch) | |
tree | b6985f45efe8864675d172487ce8d3503b8fe7b7 /src | |
parent | 8d001d40f4d45e9df16f913885d7f6dbaa179c0f (diff) | |
download | armnn-8fccd86b95be418d1b0397bfd5f48a319f210180.tar.gz |
IVGCVSW-3743 Add end-to-end test for ABS on CpuRef
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: I807a57af6dc2555a5a422388558a02e958ed610d
Diffstat (limited to 'src')
-rw-r--r-- | src/backends/backendsCommon/test/AbsEndToEndTestImpl.hpp | 65 | ||||
-rw-r--r-- | src/backends/backendsCommon/test/CMakeLists.txt | 1 | ||||
-rw-r--r-- | src/backends/reference/test/RefEndToEndTests.cpp | 18 |
3 files changed, 84 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/test/AbsEndToEndTestImpl.hpp b/src/backends/backendsCommon/test/AbsEndToEndTestImpl.hpp new file mode 100644 index 0000000000..c46376b80e --- /dev/null +++ b/src/backends/backendsCommon/test/AbsEndToEndTestImpl.hpp @@ -0,0 +1,65 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#pragma once + +#include "CommonTestUtils.hpp" + +#include <ResolveType.hpp> + +#include <armnn/ArmNN.hpp> + +namespace +{ + +armnn::INetworkPtr CreateAbsNetwork(const armnn::TensorInfo& tensorInfo) +{ + armnn::INetworkPtr network(armnn::INetwork::Create()); + + armnn::IConnectableLayer* inputLayer = network->AddInputLayer(0, "input"); + armnn::IConnectableLayer* absLayer = network->AddAbsLayer("abs"); + armnn::IConnectableLayer* outputLayer = network->AddOutputLayer(0, "output"); + + Connect(inputLayer, absLayer, tensorInfo, 0, 0); + Connect(absLayer, outputLayer, tensorInfo, 0, 0); + + return network; +} + +} // anonymous namespace + +template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> +void AbsEndToEnd(const std::vector<armnn::BackendId>& backends) +{ + using namespace armnn; + + const float qScale = IsQuantizedType<T>() ? 0.25f : 1.0f; + const int32_t qOffset = IsQuantizedType<T>() ? 50 : 0; + + TensorInfo tensorInfo({ 1, 1, 2, 3 }, ArmnnType, qScale, qOffset); + + std::vector<float> inputData = + { + -1.f, 2.f, -3.f, + 4.f, -5.f, 6.f + }; + + std::vector<float> expectedOutputData = + { + 1.f, 2.f, 3.f, + 4.f, 5.f, 6.f + }; + + // quantize data + std::vector<T> qInputData = QuantizedVector<T>(qScale, qOffset, inputData); + std::vector<T> qExpectedOutputData = QuantizedVector<T>(qScale, qOffset, expectedOutputData); + + INetworkPtr network = CreateAbsNetwork(tensorInfo); + + EndToEndLayerTestImpl<ArmnnType, ArmnnType>(std::move(network), + { { 0, qInputData } }, + { { 0, qExpectedOutputData } }, + backends); +} diff --git a/src/backends/backendsCommon/test/CMakeLists.txt b/src/backends/backendsCommon/test/CMakeLists.txt index 6b715226a2..49604b0d3b 100644 --- a/src/backends/backendsCommon/test/CMakeLists.txt +++ b/src/backends/backendsCommon/test/CMakeLists.txt @@ -4,6 +4,7 @@ # list(APPEND armnnBackendsCommonUnitTests_sources + AbsEndToEndTestImpl.hpp ActivationFixture.hpp BackendIdTests.cpp BackendRegistryTests.cpp diff --git a/src/backends/reference/test/RefEndToEndTests.cpp b/src/backends/reference/test/RefEndToEndTests.cpp index 8cca3bc1a6..6024f1584b 100644 --- a/src/backends/reference/test/RefEndToEndTests.cpp +++ b/src/backends/reference/test/RefEndToEndTests.cpp @@ -5,6 +5,7 @@ #include <backendsCommon/test/EndToEndTestImpl.hpp> +#include <backendsCommon/test/AbsEndToEndTestImpl.hpp> #include <backendsCommon/test/ArithmeticTestImpl.hpp> #include <backendsCommon/test/BatchToSpaceNdEndToEndTestImpl.hpp> #include <backendsCommon/test/ConcatEndToEndTestImpl.hpp> @@ -24,6 +25,23 @@ BOOST_AUTO_TEST_SUITE(RefEndToEnd) std::vector<armnn::BackendId> defaultBackends = {armnn::Compute::CpuRef}; +// Abs +BOOST_AUTO_TEST_CASE(RefAbsEndToEndTestFloat32) +{ + AbsEndToEnd<armnn::DataType::Float32>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefAbsEndToEndTestUint8) +{ + AbsEndToEnd<armnn::DataType::QuantisedAsymm8>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefAbsEndToEndTestInt16) +{ + AbsEndToEnd<armnn::DataType::QuantisedSymm16>(defaultBackends); +} + +// Constant BOOST_AUTO_TEST_CASE(ConstantUsage_Ref_Float32) { BOOST_TEST(ConstantUsageFloat32Test(defaultBackends)); |