commit 1d6b731aa3c56f9ff6d6fb337576ae2f1d2e3b6f
author Teresa Charlin <teresa.charlinreyes@arm.com> Wed Feb 07 22:02:48 2024 +0000
committer TeresaARM <teresa.charlinreyes@arm.com> Thu Feb 08 09:55:14 2024 +0000
tree 915ea3d14c01848555723adb63ce89ddb20b404c
parent bc5a5d59e07d2365c2c88b3430b4a311d186ebbc

IVGCVSW-7625 GpuFsa Op: Add Resize/Scale operator

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

src/backends/gpuFsa/GpuFsaLayerSupport.cpp

diff --git a/src/backends/gpuFsa/GpuFsaLayerSupport.cpp b/src/backends/gpuFsa/GpuFsaLayerSupport.cpp
index d9d95e5..56af9c4 100644
--- a/src/backends/gpuFsa/GpuFsaLayerSupport.cpp
+++ b/src/backends/gpuFsa/GpuFsaLayerSupport.cpp
@@ -16,6 +16,7 @@
 #include "layers/GpuFsaElementwiseBinaryAdd.hpp"
 #include "layers/GpuFsaElementwiseBinarySub.hpp"
 #include "layers/GpuFsaPooling2d.hpp"
+#include "layers/GpuFsaResize.hpp"
 #endif
 
 #include <vector>
@@ -187,6 +188,21 @@
                                         infos[0],
                                         *desc);
         }
+        case LayerType::Resize:
+        {
+            if (infos.size() != 2)
+            {
+                throw InvalidArgumentException("Invalid number of Resize TensorInfos. "
+                                               "TensorInfos should be of format: {input, output}.");
+            }
+
+            auto desc = PolymorphicDowncast<const ResizeDescriptor*>(&descriptor);
+
+            FORWARD_LAYER_VALIDATE_FUNC(GpuFsaResizeValidate,
+                                        reasonIfUnsupported,
+                                        infos[0],
+                                        *desc);
+        }
         case LayerType::Constant:
         case LayerType::Input:
         case LayerType::Output: