commit a7a12f5c3654da554ad6197beff0f0fc54681c92
author Matthew Sloyan <matthew.sloyan@arm.com> Thu May 06 10:05:28 2021 +0100
committer Jim Flynn <jim.flynn@arm.com> Thu May 06 14:38:18 2021 +0000
tree 52e72b87ff434af05d7f2e630b50568ed7809d29
parent cae45686aeed0761ff2c9115ef0a064278ae75fa

IVGCVSW-5969 TfLiteDelegate: Add PACK operator Support

 * Added support for PACK which is equivalent to Arm NN STACK

Signed-off-by: Matthew Sloyan <matthew.sloyan@arm.com>
Change-Id: I9ea134d0310eeea1caba30a8b9221712e9487c75

delegate/src/armnn_delegate.cpp

diff --git a/delegate/src/armnn_delegate.cpp b/delegate/src/armnn_delegate.cpp
index e637ca6..d777eff 100644
--- a/delegate/src/armnn_delegate.cpp
+++ b/delegate/src/armnn_delegate.cpp
@@ -21,6 +21,7 @@
 #include "LogicalBinary.hpp"
 #include "Lstm.hpp"
 #include "Normalization.hpp"
+#include "Pack.hpp"
 #include "Pad.hpp"
 #include "Pooling.hpp"
 #include "Quantization.hpp"
@@ -712,6 +713,12 @@
                                            tfLiteNode,
                                            nodeIndex,
                                            kTfLiteBuiltinNotEqual);
+        case kTfLiteBuiltinPack:
+            return VisitPackOperator(delegateData,
+                                     tfLiteContext,
+                                     tfLiteNode,
+                                     nodeIndex,
+                                     kTfLiteBuiltinPack);
         case kTfLiteBuiltinPad:
             return VisitPadOperator(delegateData,
                                     tfLiteContext,