diff options
author | Matthew Sloyan <matthew.sloyan@arm.com> | 2021-05-06 10:05:28 +0100 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2021-05-06 14:38:18 +0000 |
commit | a7a12f5c3654da554ad6197beff0f0fc54681c92 (patch) | |
tree | 52e72b87ff434af05d7f2e630b50568ed7809d29 /delegate/src/armnn_delegate.cpp | |
parent | cae45686aeed0761ff2c9115ef0a064278ae75fa (diff) | |
download | armnn-a7a12f5c3654da554ad6197beff0f0fc54681c92.tar.gz |
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
Diffstat (limited to 'delegate/src/armnn_delegate.cpp')
-rw-r--r-- | delegate/src/armnn_delegate.cpp | 7 |
1 files changed, 7 insertions, 0 deletions
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, |