commit d5c0ed24ce91ee0da1dcb5858da16f0f8a3d3172
author Teresa Charlin <teresa.charlinreyes@arm.com> Mon Apr 25 18:23:41 2022 +0100
committer TeresaARM <teresa.charlinreyes@arm.com> Wed May 04 12:25:18 2022 +0000
tree 4e05c08d5c224e7df9f7ea358e2c598c6be18ff9
parent 91a53eab529d88f78572b1155bfd07eb5de141f4

IVGCVSW-6858 Add GATHERNd Support to the TfLite Delegate


Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I56418875b3bb2ae45b5c69bfeaafa1a6126b8085

delegate/src/armnn_delegate.cpp

diff --git a/delegate/src/armnn_delegate.cpp b/delegate/src/armnn_delegate.cpp
index 03db4a1..4d71f26 100644
--- a/delegate/src/armnn_delegate.cpp
+++ b/delegate/src/armnn_delegate.cpp
@@ -18,6 +18,7 @@
 #include "Fill.hpp"
 #include "FullyConnected.hpp"
 #include "Gather.hpp"
+#include "GatherNd.hpp"
 #include "LogicalBinary.hpp"
 #include "Lstm.hpp"
 #include "Normalization.hpp"
@@ -635,6 +636,12 @@
                                        tfLiteNode,
                                        nodeIndex,
                                        kTfLiteBuiltinGather);
+        case kTfLiteBuiltinGatherNd:
+            return VisitGatherNdOperator(delegateData,
+                                         tfLiteContext,
+                                         tfLiteNode,
+                                         nodeIndex,
+                                         kTfLiteBuiltinGatherNd);
         case kTfLiteBuiltinGreater:
             return VisitComparisonOperator(delegateData,
                                            tfLiteContext,