commit 8ab2d7a5b69fd4c0fabd2f134bcc333512e33216
author Kevin May <kevin.may@arm.com> Fri May 07 09:32:51 2021 +0100
committer Narumol Prangnawarat <narumol.prangnawarat@arm.com> Mon May 10 09:44:38 2021 +0000
tree de90303ba517f3211c70e77a834a732109c6cc16
parent e5f0b2409c2e557a5a78e2f4659d203154289b23

IVGCVSW-5970 TfLiteDelegate: Add UNPACK operator Support


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

delegate/src/armnn_delegate.cpp

diff --git a/delegate/src/armnn_delegate.cpp b/delegate/src/armnn_delegate.cpp
index d777eff..7e4f5b5 100644
--- a/delegate/src/armnn_delegate.cpp
+++ b/delegate/src/armnn_delegate.cpp
@@ -34,6 +34,7 @@
 #include "SpaceDepth.hpp"
 #include "Split.hpp"
 #include "Transpose.hpp"
+#include "Unpack.hpp"
 
 #include <flatbuffers/flatbuffers.h>
 #include <tensorflow/lite/context_util.h>
@@ -881,6 +882,12 @@
                                            tfLiteNode,
                                            nodeIndex,
                                            kTfLiteBuiltinTanh);
+        case kTfLiteBuiltinUnpack:
+            return VisitUnpackOperator(delegateData,
+                                       tfLiteContext,
+                                       tfLiteNode,
+                                       nodeIndex,
+                                       kTfLiteBuiltinUnpack);
         default:
             return kTfLiteError;
     }