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/backends/backendsCommon/test/AbsEndToEndTestImpl.hpp | |
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/backends/backendsCommon/test/AbsEndToEndTestImpl.hpp')
-rw-r--r-- | src/backends/backendsCommon/test/AbsEndToEndTestImpl.hpp | 65 |
1 files changed, 65 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); +} |