From bc3924503cf7838eea7e9fcca4a22e2bf54ba97f Mon Sep 17 00:00:00 2001
From: David Beck <david.beck@arm.com>
Date: Mon, 10 Sep 2018 14:47:28 +0100
Subject: IVGCVSW-1802 : Neon Sub Workload

Change-Id: I1e914b046cd8fd25669390d477f101098fe0d476
---
 src/armnn/backends/NeonLayerSupport.cpp | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

(limited to 'src/armnn/backends/NeonLayerSupport.cpp')

diff --git a/src/armnn/backends/NeonLayerSupport.cpp b/src/armnn/backends/NeonLayerSupport.cpp
index 7f33c48ed1..30956dfba0 100644
--- a/src/armnn/backends/NeonLayerSupport.cpp
+++ b/src/armnn/backends/NeonLayerSupport.cpp
@@ -27,6 +27,7 @@
 #include "NeonWorkloads/NeonPermuteWorkload.hpp"
 #include "NeonWorkloads/NeonPooling2dBaseWorkload.hpp"
 #include "NeonWorkloads/NeonSoftmaxBaseWorkload.hpp"
+#include "NeonWorkloads/NeonSubtractionFloatWorkload.hpp"
 #endif
 
 using namespace boost;
@@ -239,8 +240,11 @@ bool IsSubtractionSupportedNeon(const TensorInfo& input0,
                                 const TensorInfo& output,
                                 std::string* reasonIfUnsupported)
 {
-    // At the moment subtraction is not supported
-    return false;
+    FORWARD_WORKLOAD_VALIDATE_FUNC(NeonSubtractionWorkloadValidate,
+                                   reasonIfUnsupported,
+                                   input0,
+                                   input1,
+                                   output);
 }
 
 bool IsFullyConnectedSupportedNeon(const TensorInfo& input,
-- 
cgit v1.2.1