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

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));