commit 34fa1bd7994af9abf52dbcc4aa808d0fa5f14aa3
author Sadik Armagan <sadik.armagan@arm.com> Fri Nov 27 12:40:52 2020 +0000
committer TeresaARM <teresa.charlinreyes@arm.com> Mon Nov 30 15:12:22 2020 +0000
tree 8ff6041b92da57cd56ccbcd3b746bdc0fb29078a
parent b7fa5104715e097bf3778d2485d759a4612460cc

IVGCVSW-5393 'TfLiteDelegate: Implement the split operators'

* Added SPLIT and SPLIT_V support to armnn_delegate

Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I2def9b8be783b25ef17a997e521c6027553035d3

delegate/src/armnn_delegate.cpp

diff --git a/delegate/src/armnn_delegate.cpp b/delegate/src/armnn_delegate.cpp
index d2b1796..9097211 100644
--- a/delegate/src/armnn_delegate.cpp
+++ b/delegate/src/armnn_delegate.cpp
@@ -27,6 +27,7 @@
 #include "Slice.hpp"
 #include "Softmax.hpp"
 #include "SpaceDepth.hpp"
+#include "Split.hpp"
 #include "Transpose.hpp"
 
 #include <flatbuffers/flatbuffers.h>
@@ -714,6 +715,18 @@
                                                  tfLiteNode,
                                                  nodeIndex,
                                                  armnn::UnaryOperation::Rsqrt);
+        case kTfLiteBuiltinSplit:
+            return VisitSplitOperator(delegateData,
+                                      tfLiteContext,
+                                      tfLiteNode,
+                                      nodeIndex,
+                                      kTfLiteBuiltinSplit);
+        case kTfLiteBuiltinSplitV:
+            return VisitSplitVOperator(delegateData,
+                                       tfLiteContext,
+                                       tfLiteNode,
+                                       nodeIndex,
+                                       kTfLiteBuiltinSplitV);
         case kTfLiteBuiltinSqrt:
             return VisitElementwiseUnaryOperator(delegateData,
                                                  tfLiteContext,