aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/test/ReverseV2EndToEndTestImpl.hpp
diff options
context:
space:
mode:
authorDeclan-ARM <decmce01@arm.com>2023-07-20 17:32:57 +0100
committerNikhil Raj Arm <nikhil.raj@arm.com>2023-07-26 13:07:52 +0000
commit1bf56cde4f199ea8df722d1875ee44bd0ffb414a (patch)
treee13430cac7882490ec47cc88a728398822deb9c5 /src/backends/backendsCommon/test/ReverseV2EndToEndTestImpl.hpp
parente86aa90fee62970e63bb0239c6a3952f06b8d2c2 (diff)
downloadarmnn-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.hpp72
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
+