From 9145e38edf49fa4862008c163c34590141eecb14 Mon Sep 17 00:00:00 2001
From: Teresa Charlin <teresa.charlinreyes@arm.com>
Date: Thu, 17 Aug 2023 18:44:58 +0100
Subject: IVGCVSW-7505 Create FusedLayer and NeonFusedWorkload for AddMulAdd
 Neon kernel

Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Ic778d35b001474b44fb1e433a6fe276e4ec9f565
---
 src/backends/backendsCommon/WorkloadFactory.cpp | 30 +++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

(limited to 'src/backends/backendsCommon/WorkloadFactory.cpp')

diff --git a/src/backends/backendsCommon/WorkloadFactory.cpp b/src/backends/backendsCommon/WorkloadFactory.cpp
index f067056ce1..6ff237bc12 100644
--- a/src/backends/backendsCommon/WorkloadFactory.cpp
+++ b/src/backends/backendsCommon/WorkloadFactory.cpp
@@ -525,6 +525,36 @@ bool IWorkloadFactory::IsLayerConfigurationSupported(const BackendId& backendId,
                                                reason);
             break;
         }
+        case LayerType::Fused:
+        {
+            auto cLayer = PolymorphicDowncast<const FusedLayer*>(&layer);
+
+            // Get vector of all outputs.
+            auto getOutTensorInfo = [&dataType](const OutputSlot& slot)
+            {
+                return OverrideDataType(slot.GetTensorInfo(), dataType);
+            };
+            auto beginOutputs = MakeTransformIterator(layer.GetOutputSlots().begin(), getOutTensorInfo);
+            auto endOutputs = MakeTransformIterator(layer.GetOutputSlots().end(), getOutTensorInfo);
+            std::vector<TensorInfo> outputs(beginOutputs, endOutputs);
+            const std::vector<std::reference_wrapper<TensorInfo>> outputPtrs(outputs.begin(), outputs.end());
+
+            // Get vector of all inputs.
+            auto getInputTensorInfo = [&dataType](const InputSlot& slot)
+            {
+                return OverrideDataType(slot.GetTensorInfo(), dataType);
+            };
+            auto beginInputs = MakeTransformIterator(layer.GetInputSlots().begin(), getInputTensorInfo);
+            auto endInputs = MakeTransformIterator(layer.GetInputSlots().end(), getInputTensorInfo);
+            std::vector<TensorInfo> inputs(beginInputs, endInputs);
+            const std::vector<std::reference_wrapper<TensorInfo>> inputPtrs(inputs.begin(), inputs.end());
+
+            result = layerSupportObject.IsFusedSupported(inputPtrs,
+                                                         outputPtrs,
+                                                         cLayer->GetParameters(),
+                                                         reason);
+            break;
+        }
         case LayerType::Gather:
         {
             const TensorInfo& input0 = layer.GetInputSlot(0).GetTensorInfo();
-- 
cgit v1.2.1