commit 829e13edbe9487b5a0600688cec6312b867e2f07
author John Mcloughlin <john.mcloughlin@arm.com> Wed Jan 31 11:00:27 2024 +0000
committer John Mcloughlin <john.mcloughlin@arm.com> Wed Jan 31 21:24:23 2024 +0000
tree 2b9176173e73afc1f1cdf62cec334c7ee64e7123
parent e4134fc28ce2a9c751fb22ff84b733d067255849

IVGCVSW-7568 Implement Sub ElementwiseBinary operator GpuFsa

* Added support for Gpu Sub operator
* Added unit tests

Signed-off-by: John Mcloughlin <john.mcloughlin@arm.com>
Change-Id: I1efaa485772a3716e3781566843bd50bd9bab811

src/backends/gpuFsa/GpuFsaBackend.cpp

diff --git a/src/backends/gpuFsa/GpuFsaBackend.cpp b/src/backends/gpuFsa/GpuFsaBackend.cpp
index 7951b17..8b62aec 100644
--- a/src/backends/gpuFsa/GpuFsaBackend.cpp
+++ b/src/backends/gpuFsa/GpuFsaBackend.cpp
@@ -23,6 +23,7 @@
 #include "layers/GpuFsaConvolution2d.hpp"
 #include "layers/GpuFsaDepthwiseConvolution2d.hpp"
 #include "layers/GpuFsaElementwiseBinaryAdd.hpp"
+#include "layers/GpuFsaElementwiseBinarySub.hpp"
 
 namespace armnn
 {
@@ -305,6 +306,13 @@
 
                     GpuFsaElementwiseBinaryAddCreateOp(preCompiledBlobPtr, input0, input1);
                 }
+                else if (desc->m_Operation == BinaryOperation::Sub)
+                {
+                    auto input0 = base.GetInputSlot(0).GetConnectedOutputSlot()->GetTensorInfo();
+                    auto input1 = base.GetInputSlot(1).GetConnectedOutputSlot()->GetTensorInfo();
+
+                    GpuFsaElementwiseBinarySubCreateOp(preCompiledBlobPtr, input0, input1);
+                }
                 break;
             }
             default: