commit cbf79298f73310fe4ca5d760ded73575e4bf8fad
author Idriss Chaouch <idriss.chaouch@arm.com> Fri Sep 08 11:18:16 2023 +0100
committer Idriss Chaouch <idriss.chaouch@arm.com> Tue Sep 12 11:18:53 2023 +0100
tree 95d1c85c20a6af48427ea50721d65ea6e72cd10d
parent cd0874c5e404f8676d988f7d241aa1f88a037dd2

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

diff --git a/delegate/classic/src/armnn_delegate.cpp b/delegate/classic/src/armnn_delegate.cpp
index de2aa0c..c428d46 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 @@
                                                tfLiteNode,
                                                nodeIndex,
                                                kTfLiteBuiltinBatchToSpaceNd);
+        case kTfLiteBuiltinBroadcastTo:
+            return VisitBroadcastToOperator(delegateData,
+                                            tfLiteContext,
+                                            tfLiteNode,
+                                            nodeIndex,
+                                            kTfLiteBuiltinBroadcastTo);
         case kTfLiteBuiltinCast:
             return VisitCastOperator(delegateData,
                                      tfLiteContext,