From f195f03e095a5c4dc6880be11af64cab83b5c94b Mon Sep 17 00:00:00 2001
From: David Beck <david.beck@arm.com>
Date: Thu, 6 Sep 2018 16:46:34 +0100
Subject: IVGCVSW-1803 : add Ref Subtraction layer

Change-Id: I4c019d626f9369245eca6d549bbe7a28e141f198
---
 src/armnn/backends/test/LayerTests.hpp | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

(limited to 'src/armnn/backends/test/LayerTests.hpp')

diff --git a/src/armnn/backends/test/LayerTests.hpp b/src/armnn/backends/test/LayerTests.hpp
index 06d789e783..5ca4c49c88 100644
--- a/src/armnn/backends/test/LayerTests.hpp
+++ b/src/armnn/backends/test/LayerTests.hpp
@@ -185,7 +185,11 @@ LayerTestResult<float, 4> AdditionBroadcast1ElementTest(armnn::IWorkloadFactory&
 LayerTestResult<float, 4> AdditionBroadcastTest(armnn::IWorkloadFactory& workloadFactory);
 
 LayerTestResult<float, 4> CompareAdditionTest(armnn::IWorkloadFactory& workloadFactory,
-                                       armnn::IWorkloadFactory& refWorkloadFactory);
+                                              armnn::IWorkloadFactory& refWorkloadFactory);
+
+LayerTestResult<float, 4> SubtractionTest(armnn::IWorkloadFactory& workloadFactory);
+LayerTestResult<float, 4> SubtractionBroadcast1ElementTest(armnn::IWorkloadFactory& workloadFactory);
+LayerTestResult<float, 4> SubtractionBroadcastTest(armnn::IWorkloadFactory& workloadFactory);
 
 LayerTestResult<float, 4> CompareActivationTest(armnn::IWorkloadFactory&  workloadFactory,
                                                 armnn::IWorkloadFactory&  refWorkloadFactory,
@@ -264,6 +268,10 @@ LayerTestResult<uint8_t, 4> AdditionUint8Test(armnn::IWorkloadFactory& workloadF
 LayerTestResult<uint8_t, 4> AdditionBroadcast1ElementUint8Test(armnn::IWorkloadFactory& workloadFactory);
 LayerTestResult<uint8_t, 4> AdditionBroadcastUint8Test(armnn::IWorkloadFactory& workloadFactory);
 
+LayerTestResult<uint8_t, 4> SubtractionUint8Test(armnn::IWorkloadFactory& workloadFactory);
+LayerTestResult<uint8_t, 4> SubtractionBroadcast1ElementUint8Test(armnn::IWorkloadFactory& workloadFactory);
+LayerTestResult<uint8_t, 4> SubtractionBroadcastUint8Test(armnn::IWorkloadFactory& workloadFactory);
+
 LayerTestResult<uint8_t, 4> CompareActivationUint8Test(armnn::IWorkloadFactory&  workloadFactory,
                                                        armnn::IWorkloadFactory&  refWorkloadFactory,
                                                        armnn::ActivationFunction f);
-- 
cgit v1.2.1