diff options
author | Declan-ARM <decmce01@arm.com> | 2023-07-20 17:32:57 +0100 |
---|---|---|
committer | Nikhil Raj Arm <nikhil.raj@arm.com> | 2023-07-26 13:07:52 +0000 |
commit | 1bf56cde4f199ea8df722d1875ee44bd0ffb414a (patch) | |
tree | e13430cac7882490ec47cc88a728398822deb9c5 /src/backends/backendsCommon/test/ReverseV2EndToEndTestImpl.hpp | |
parent | e86aa90fee62970e63bb0239c6a3952f06b8d2c2 (diff) | |
download | armnn-1bf56cde4f199ea8df722d1875ee44bd0ffb414a.tar.gz |
IVGCVSW-7836 Add ReverseV2 End-to-End Tests
* create implementation header
* add copyright notice
* add pragma once and anonymous namespace
* create network function declaration
* complete body of network function
* create end-to-end function declaration
* complete body of end-to-end function
* add references to tests for supported data types
Signed-off-by: Declan-ARM <decmce01@arm.com>
Signed-off-by: Matthew Sloyan <matthew.sloyan@arm.com>
Change-Id: I376e48efd8b6ca9e0e0b05b516be599c0acdbd16
Diffstat (limited to 'src/backends/backendsCommon/test/ReverseV2EndToEndTestImpl.hpp')
-rw-r--r-- | src/backends/backendsCommon/test/ReverseV2EndToEndTestImpl.hpp | 72 |
1 files changed, 72 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/test/ReverseV2EndToEndTestImpl.hpp b/src/backends/backendsCommon/test/ReverseV2EndToEndTestImpl.hpp new file mode 100644 index 0000000000..b36d2eeb59 --- /dev/null +++ b/src/backends/backendsCommon/test/ReverseV2EndToEndTestImpl.hpp @@ -0,0 +1,72 @@ +// +// Copyright © 2023 Arm Ltd and Contributors. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#pragma once + +namespace +{ + +armnn::INetworkPtr CreateReverseV2Network(const armnn::TensorInfo& inputInfo, + const armnn::TensorInfo& axisInfo, + const armnn::TensorInfo& outputInfo, + const std::vector<int32_t>& axisData) +{ + using namespace armnn; + + INetworkPtr network(INetwork::Create()); + + IConnectableLayer* inputLayer = network->AddInputLayer(0, "input"); + IConnectableLayer* axisLayer = network->AddConstantLayer(ConstTensor(axisInfo, axisData)); + IConnectableLayer* reverseLayer = network->AddReverseV2Layer("reverseV2"); + IConnectableLayer* outputLayer = network->AddOutputLayer(0, "output"); + + Connect(inputLayer, reverseLayer, inputInfo, 0, 0); + Connect(axisLayer, reverseLayer, axisInfo, 0, 1); + Connect(reverseLayer, outputLayer, outputInfo, 0, 0); + + return network; +} + +template <armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> +void ReverseV2EndToEnd(const std::vector<BackendId>& backends) +{ + float qScale = 1.0f; + int32_t qOffset = 0; + bool qConst = true; + + TensorInfo inputInfo ( { 2, 3, 4 }, ArmnnType, qScale, qOffset, qConst); + TensorInfo axisInfo ( { 1 } , DataType::Signed32, qScale, qOffset, qConst); + TensorInfo outputInfo ( { 2, 3, 4 }, ArmnnType, qScale, qOffset); + + std::vector<T> inputData = armnnUtils::QuantizedVector<T>({ + 1, 2, 3, 4, + 5, 6 , 7, 8, + 9, 10, 11, 12, + 13, 14, 15, 16, + 17, 18, 19, 20, + 21, 22, 23, 24 + }, qScale, qOffset); + + std::vector<int32_t> axisData = { 1 }; + + std::vector<T> expectedOutputData = armnnUtils::QuantizedVector<T>({ + 9, 10, 11, 12, + 5, 6, 7 , 8, + 1, 2, 3, 4, + 21, 22, 23, 24, + 17, 18, 19, 20, + 13, 14, 15, 16 + }, qScale, qOffset); + + INetworkPtr network = CreateReverseV2Network(inputInfo, axisInfo, outputInfo, axisData); + + std::map<int, std::vector<T>> inputTensor = { { 0, inputData } }; + std::map<int, std::vector<T>> expectedOutputTensor = { { 0, expectedOutputData } }; + + EndToEndLayerTestImpl<ArmnnType, ArmnnType>(std::move(network), inputTensor, expectedOutputTensor, backends); +} + +} // anonymous namespace + |