commit ddbda6a1ed13f7bef7e0dce07a37e91b062ce98a
author Teresa Charlin <teresa.charlinreyes@arm.com> Wed Feb 07 22:58:29 2024 +0000
committer TeresaARM <teresa.charlinreyes@arm.com> Fri Feb 09 09:59:02 2024 +0000
tree 0dca42c431988340cdc0cdc6592731fef9a068ad
parent 33753901d4c77c958d006fb8e4a283a9a33c4426

IVGCVSW-8275 GpuFsa Op: Add Activation functions available

* Currently Sigmoid and TanH Functions are implemented.

Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: If9483be9201dfe47b86acc41ec7932725ac2e39e

src/backends/gpuFsa/GpuFsaLayerSupport.cpp

diff --git a/src/backends/gpuFsa/GpuFsaLayerSupport.cpp b/src/backends/gpuFsa/GpuFsaLayerSupport.cpp
index 98fb430..d75f18c 100644
--- a/src/backends/gpuFsa/GpuFsaLayerSupport.cpp
+++ b/src/backends/gpuFsa/GpuFsaLayerSupport.cpp
@@ -10,6 +10,7 @@
 #include <armnn/utility/PolymorphicDowncast.hpp>
 
 #if defined(ARMCOMPUTEGPUFSA_ENABLED)
+#include "layers/GpuFsaActivation.hpp"
 #include "layers/GpuFsaBatchMatMul.hpp"
 #include "layers/GpuFsaCast.hpp"
 #include "layers/GpuFsaConvolution2d.hpp"
@@ -78,6 +79,20 @@
 
     switch (type)
     {
+        case LayerType::Activation:
+        {
+            if (infos.size() != 2)
+            {
+                throw InvalidArgumentException("Invalid number of Activation TensorInfos. "
+                                               "TensorInfos should be of format: {input, output}.");
+            }
+
+            auto desc = PolymorphicDowncast<const ActivationDescriptor*>(&descriptor);
+            FORWARD_LAYER_VALIDATE_FUNC(GpuFsaActivationValidate,
+                                        reasonIfUnsupported,
+                                        infos[0],
+                                        *desc);
+        }
         case LayerType::BatchMatMul:
         {
             if (infos.size() != 3)
@@ -87,7 +102,6 @@
             }
 
             auto desc = PolymorphicDowncast<const BatchMatMulDescriptor*>(&descriptor);
-
             FORWARD_LAYER_VALIDATE_FUNC(GpuFsaBatchMatMulValidate,
                                         reasonIfUnsupported,
                                         infos[0],
@@ -101,6 +115,7 @@
                 throw InvalidArgumentException("Invalid number of cast TensorInfos. "
                                                "TensorInfos should be of format: {input, output}.");
             }
+
             FORWARD_LAYER_VALIDATE_FUNC(GpuFsaCastValidate,
                                         reasonIfUnsupported,
                                         infos[0],
@@ -186,7 +201,6 @@
             }
 
             auto desc = PolymorphicDowncast<const Pooling2dDescriptor*>(&descriptor);
-
             FORWARD_LAYER_VALIDATE_FUNC(GpuFsaPooling2dValidate,
                                         reasonIfUnsupported,
                                         infos[0],
@@ -201,7 +215,6 @@
             }
 
             auto desc = PolymorphicDowncast<const ResizeDescriptor*>(&descriptor);
-
             FORWARD_LAYER_VALIDATE_FUNC(GpuFsaResizeValidate,
                                         reasonIfUnsupported,
                                         infos[0],