From cbf79298f73310fe4ca5d760ded73575e4bf8fad Mon Sep 17 00:00:00 2001
From: Idriss Chaouch <idriss.chaouch@arm.com>
Date: Fri, 8 Sep 2023 11:18:16 +0100
Subject: IVGCVSW-8037 Add BROADCAST_TO to tflite classic and opaque delegate.

Signed-off-by: Idriss Chaouch <idriss.chaouch@arm.com>
Change-Id: Ibc145d0ea1ac9414b6a68b5b547bf2ea2852fd36
---
 delegate/classic/src/armnn_delegate.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'delegate/classic/src/armnn_delegate.cpp')

diff --git a/delegate/classic/src/armnn_delegate.cpp b/delegate/classic/src/armnn_delegate.cpp
index de2aa0c632..c428d46d87 100644
--- a/delegate/classic/src/armnn_delegate.cpp
+++ b/delegate/classic/src/armnn_delegate.cpp
@@ -11,6 +11,7 @@
 #include "ArgMinMax.hpp"
 #include "BatchMatMul.hpp"
 #include "BatchSpace.hpp"
+#include "BroadcastTo.hpp"
 #include "Comparison.hpp"
 #include "Convolution.hpp"
 #include "Control.hpp"
@@ -603,6 +604,12 @@ TfLiteStatus ArmnnSubgraph::VisitNode(DelegateData& delegateData,
                                                tfLiteNode,
                                                nodeIndex,
                                                kTfLiteBuiltinBatchToSpaceNd);
+        case kTfLiteBuiltinBroadcastTo:
+            return VisitBroadcastToOperator(delegateData,
+                                            tfLiteContext,
+                                            tfLiteNode,
+                                            nodeIndex,
+                                            kTfLiteBuiltinBroadcastTo);
         case kTfLiteBuiltinCast:
             return VisitCastOperator(delegateData,
                                      tfLiteContext,
-- 
cgit v1.2.1