commit 8fccd86b95be418d1b0397bfd5f48a319f210180
author Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> Mon Sep 09 11:22:56 2019 +0100
committer Kevin May <kevin.may@arm.com> Mon Sep 09 15:18:08 2019 +0000
tree b6985f45efe8864675d172487ce8d3503b8fe7b7
parent 8d001d40f4d45e9df16f913885d7f6dbaa179c0f

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

src/backends/backendsCommon/test/AbsEndToEndTestImpl.hpp

diff --git a/src/backends/backendsCommon/test/AbsEndToEndTestImpl.hpp b/src/backends/backendsCommon/test/AbsEndToEndTestImpl.hpp
new file mode 100644
index 0000000..c46376b
--- /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);
+}

src/backends/backendsCommon/test/CMakeLists.txt

diff --git a/src/backends/backendsCommon/test/CMakeLists.txt b/src/backends/backendsCommon/test/CMakeLists.txt
index 6b71522..49604b0 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

src/backends/reference/test/RefEndToEndTests.cpp

diff --git a/src/backends/reference/test/RefEndToEndTests.cpp b/src/backends/reference/test/RefEndToEndTests.cpp
index 8cca3bc..6024f15 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 @@
 
 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));