From a7a12f5c3654da554ad6197beff0f0fc54681c92 Mon Sep 17 00:00:00 2001
From: Matthew Sloyan <matthew.sloyan@arm.com>
Date: Thu, 6 May 2021 10:05:28 +0100
Subject: 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 | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'delegate/src/armnn_delegate.cpp')

diff --git a/delegate/src/armnn_delegate.cpp b/delegate/src/armnn_delegate.cpp
index e637ca6ebf..d777eff27c 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 @@ TfLiteStatus ArmnnSubgraph::VisitNode(DelegateData& delegateData,
                                            tfLiteNode,
                                            nodeIndex,
                                            kTfLiteBuiltinNotEqual);
+        case kTfLiteBuiltinPack:
+            return VisitPackOperator(delegateData,
+                                     tfLiteContext,
+                                     tfLiteNode,
+                                     nodeIndex,
+                                     kTfLiteBuiltinPack);
         case kTfLiteBuiltinPad:
             return VisitPadOperator(delegateData,
                                     tfLiteContext,
-- 
cgit v1.2.1