From aac61129e4317000c46e37063b010851b0e17d27 Mon Sep 17 00:00:00 2001 From: Teresa Charlin Date: Thu, 5 May 2022 16:11:36 +0100 Subject: IVGCVSW-6936 Add SQRT support to CL Signed-off-by: Teresa Charlin Change-Id: Ib90bade63cd0437329c690b09cf719a2e2bd06a4 --- src/backends/cl/ClLayerSupport.cpp | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'src/backends/cl/ClLayerSupport.cpp') diff --git a/src/backends/cl/ClLayerSupport.cpp b/src/backends/cl/ClLayerSupport.cpp index 09b3f43838..6b8cf52d87 100644 --- a/src/backends/cl/ClLayerSupport.cpp +++ b/src/backends/cl/ClLayerSupport.cpp @@ -74,6 +74,7 @@ #include "workloads/ClSpaceToBatchNdWorkload.hpp" #include "workloads/ClSpaceToDepthWorkload.hpp" #include "workloads/ClSplitterWorkload.hpp" +#include "workloads/ClSqrtWorkload.hpp" #include "workloads/ClStackWorkload.hpp" #include "workloads/ClStridedSliceWorkload.hpp" #include "workloads/ClSubtractionWorkload.hpp" @@ -957,6 +958,11 @@ bool ClLayerSupport::IsElementwiseUnarySupported(const TensorInfo& input, reasonIfUnsupported, input, output); + case UnaryOperation::Sqrt: + FORWARD_WORKLOAD_VALIDATE_FUNC(ClSqrtWorkloadValidate, + reasonIfUnsupported, + input, + output); default: return false; } -- cgit v1.2.1