commit ac47210621a601a2b6ded78e003d136626434f50
author Sadik Armagan <sadik.armagan@arm.com> Tue Mar 24 09:54:36 2020 +0000
committer Sadik Armagan <sadik.armagan@arm.com> Tue Mar 24 14:28:53 2020 +0000
tree 7bb4940dd25ac2dac85c97227b82d567bd2a4d87
parent 957b844f130066f00c8adbafae36cec2ab3e9548

IVGCVSW-3813 Add Unary Elementwise Operation 'NEG' support to the android-nn-driver

* Implemented ClNegWorkload
* Implemented NeonNegWorkload
* Enabled 'NEG' operator on CL and Neon as well as Ref
* Implemented unit tests for 'NEG' operator

Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I3d7a892692716636cae6bdf8ddd238e3d1ea064f

src/backends/neon/NeonLayerSupport.cpp

diff --git a/src/backends/neon/NeonLayerSupport.cpp b/src/backends/neon/NeonLayerSupport.cpp
index 7877612..999b158 100644
--- a/src/backends/neon/NeonLayerSupport.cpp
+++ b/src/backends/neon/NeonLayerSupport.cpp
@@ -38,6 +38,7 @@
 #include "workloads/NeonMinimumWorkload.hpp"
 #include "workloads/NeonMultiplicationWorkload.hpp"
 #include "workloads/NeonDivisionWorkload.hpp"
+#include "workloads/NeonNegWorkload.hpp"
 #include "workloads/NeonNormalizationFloatWorkload.hpp"
 #include "workloads/NeonFullyConnectedWorkload.hpp"
 #include "workloads/NeonPadWorkload.hpp"
@@ -370,6 +371,13 @@
                                        input,
                                        output);
     }
+    else if (descriptor.m_Operation == UnaryOperation::Neg)
+    {
+        FORWARD_WORKLOAD_VALIDATE_FUNC(NeonNegWorkloadValidate,
+                                       reasonIfUnsupported,
+                                       input,
+                                       output);
+    }
 
     return false;
 }