From 98e383eadf4e670d057ad725c7fe7924fea8e36b Mon Sep 17 00:00:00 2001
From: Idriss Chaouch <idriss.chaouch@arm.com>
Date: Mon, 28 Aug 2023 14:28:31 +0100
Subject: IVGCVSW-7525 Add broadcast_to operator

Signed-off-by: Idriss Chaouch <idriss.chaouch@arm.com>
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I94ec5f9120b2d736fdf98d00ec5137a4efd739b8
---
 src/backends/backendsCommon/WorkloadFactory.cpp | 12 ++++++++++++
 1 file changed, 12 insertions(+)

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

diff --git a/src/backends/backendsCommon/WorkloadFactory.cpp b/src/backends/backendsCommon/WorkloadFactory.cpp
index 6ff237bc12..2538211a41 100644
--- a/src/backends/backendsCommon/WorkloadFactory.cpp
+++ b/src/backends/backendsCommon/WorkloadFactory.cpp
@@ -183,6 +183,18 @@ bool IWorkloadFactory::IsLayerConfigurationSupported(const BackendId& backendId,
                                                                   reason);
             break;
         }
+        case LayerType::BroadcastTo:
+        {
+            auto cLayer = PolymorphicDowncast<const BroadcastToLayer*>(&layer);
+            const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
+            const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
+
+            result = layerSupportObject.IsBroadcastToSupported(OverrideDataType(input, dataType),
+                                                               OverrideDataType(output, dataType),
+                                                               cLayer->GetParameters(),
+                                                               reason);
+            break;
+        }
         case LayerType::Cast:
         {
             const TensorInfo& input = layer.GetInputSlot(0).GetTensorInfo();
-- 
cgit v1.2.1