From 06145cc4b2734608b8abe36d5bf06f7449b818ca Mon Sep 17 00:00:00 2001
From: Teresa Charlin <teresa.charlinreyes@arm.com>
Date: Thu, 5 May 2022 15:31:30 +0100
Subject: IVGCVSW-6936 Add SQRT support to Neon

Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I195957541069cb52cdd2c8aead0e4a34498a6f38
---
 src/backends/neon/NeonLayerSupport.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

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

diff --git a/src/backends/neon/NeonLayerSupport.cpp b/src/backends/neon/NeonLayerSupport.cpp
index 26b650b49d..d50b25377d 100644
--- a/src/backends/neon/NeonLayerSupport.cpp
+++ b/src/backends/neon/NeonLayerSupport.cpp
@@ -72,6 +72,7 @@
 #include "workloads/NeonSpaceToBatchNdWorkload.hpp"
 #include "workloads/NeonSpaceToDepthWorkload.hpp"
 #include "workloads/NeonSplitterWorkload.hpp"
+#include "workloads/NeonSqrtWorkload.hpp"
 #include "workloads/NeonStackWorkload.hpp"
 #include "workloads/NeonStridedSliceWorkload.hpp"
 #include "workloads/NeonSubtractionWorkload.hpp"
@@ -941,6 +942,11 @@ bool NeonLayerSupport::IsElementwiseUnarySupported(const TensorInfo& input,
                                            reasonIfUnsupported,
                                            input,
                                            output);
+        case UnaryOperation::Sqrt:
+            FORWARD_WORKLOAD_VALIDATE_FUNC(NeonSqrtWorkloadValidate,
+                                           reasonIfUnsupported,
+                                           input,
+                                           output);
         default:
             return false;
     }
-- 
cgit v1.2.1