diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2022-05-05 16:11:36 +0100 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2022-05-06 13:47:10 +0000 |
commit | aac61129e4317000c46e37063b010851b0e17d27 (patch) | |
tree | 8c4d07b391609e9c84d23261ace408c7e54b2316 /src/backends/cl/ClLayerSupport.cpp | |
parent | aa8051015a5de37438ef5526417156e51296d0f4 (diff) | |
download | armnn-aac61129e4317000c46e37063b010851b0e17d27.tar.gz |
IVGCVSW-6936 Add SQRT support to CL
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Ib90bade63cd0437329c690b09cf719a2e2bd06a4
Diffstat (limited to 'src/backends/cl/ClLayerSupport.cpp')
-rw-r--r-- | src/backends/cl/ClLayerSupport.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
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; } |