commit b2831c5d179c5853c884280107d2420904441a2c
author Kevin May <kevin.may@arm.com> Wed Apr 26 17:27:24 2023 +0100
committer Kevin May <kevin.may@arm.com> Thu Apr 27 08:43:00 2023 +0000
tree ca468711da2c5032f7c308245f1beac9286b7863
parent 559d9097ac8e95d53d507074dbb1ba69d42664a6

IVGCVSW-7591 IVGCVSW-7592 Add Gather and GatherNd to Opaque Delegate

Signed-off-by: Kevin May <kevin.may@arm.com>
Change-Id: Id2b6a4f70b1cb50e5f7f7ab4e30487b3816c9ad4

delegate/opaque/src/armnn_delegate.cpp

diff --git a/delegate/opaque/src/armnn_delegate.cpp b/delegate/opaque/src/armnn_delegate.cpp
index 4d43b92..38a67e7 100644
--- a/delegate/opaque/src/armnn_delegate.cpp
+++ b/delegate/opaque/src/armnn_delegate.cpp
@@ -670,6 +670,18 @@
                                            tfLiteNode,
                                            nodeIndex,
                                            kTfLiteBuiltinEqual);
+        case kTfLiteBuiltinGather:
+            return VisitGatherOperator(delegateData,
+                                       tfLiteContext,
+                                       tfLiteNode,
+                                       nodeIndex,
+                                       kTfLiteBuiltinGather);
+        case kTfLiteBuiltinGatherNd:
+            return VisitGatherNdOperator(delegateData,
+                                         tfLiteContext,
+                                         tfLiteNode,
+                                         nodeIndex,
+                                         kTfLiteBuiltinGatherNd);
         case kTfLiteBuiltinGreater:
             return VisitComparisonOperator(delegateData,
                                            tfLiteContext,