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/opaque/src/armnn_delegate.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

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

diff --git a/delegate/opaque/src/armnn_delegate.cpp b/delegate/opaque/src/armnn_delegate.cpp
index bad1abaa59..08b1504efb 100644
--- a/delegate/opaque/src/armnn_delegate.cpp
+++ b/delegate/opaque/src/armnn_delegate.cpp
@@ -10,6 +10,7 @@
 #include "ArgMinMax.hpp"
 #include "BatchMatMul.hpp"
 #include "BatchSpace.hpp"
+#include "BroadcastTo.hpp"
 #include "Comparison.hpp"
 #include "Convolution.hpp"
 #include "Control.hpp"
@@ -654,6 +655,12 @@ TfLiteStatus ArmnnSubgraph::VisitNode(DelegateData& delegateData,
                                             tfLiteNode,
                                             nodeIndex,
                                             kTfLiteBuiltinBatchMatmul);
+        case kTfLiteBuiltinBroadcastTo:
+            return VisitBroadcastToOperator(delegateData,
+                                            tfLiteContext,
+                                            tfLiteNode,
+                                            nodeIndex,
+                                            kTfLiteBuiltinBroadcastTo);
         case kTfLiteBuiltinBatchToSpaceNd:
             return VisitBatchToSpaceNdOperator(delegateData,
                                                tfLiteContext,
-- 
cgit v1.2.1