diff options
| author | Keith Davis <keith.davis@arm.com> | 2021-06-01 17:36:32 +0100 |
|---|---|---|
| committer | KeithARM <keith.davis@arm.com> | 2021-06-18 09:35:52 +0000 |
| commit | 0176fd81b3f6a82ddc89e016cb634010f5397425 (patch) | |
| tree | 93f140f935a9ff312276fe663279402af53f9b03 /delegate/src/armnn_delegate.cpp | |
| parent | 32b6af5d0bd85a06b3400f22a58d0eeaba04ba32 (diff) | |
| download | armnn-0176fd81b3f6a82ddc89e016cb634010f5397425.tar.gz | |
MLCE-510 Add CpuRef Shape Operator to ArmNN
* Add TfLiteParser and delegate support
Signed-off-by: Keith Davis <keith.davis@arm.com>
Change-Id: Id3219ba7cc7128b5e73de2c7d8d076a40dcce9c5
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 0c984ecc82..0ac33808b0 100644 --- a/delegate/src/armnn_delegate.cpp +++ b/delegate/src/armnn_delegate.cpp @@ -30,6 +30,7 @@ #include "Reduce.hpp" #include "Resize.hpp" #include "Round.hpp" +#include "Shape.hpp" #include "Slice.hpp" #include "Softmax.hpp" #include "SpaceDepth.hpp" @@ -805,6 +806,12 @@ TfLiteStatus ArmnnSubgraph::VisitNode(DelegateData& delegateData, tfLiteNode, nodeIndex, armnn::UnaryOperation::Rsqrt); + case kTfLiteBuiltinShape: + return VisitShapeOperator(delegateData, + tfLiteContext, + tfLiteNode, + nodeIndex, + kTfLiteBuiltinShape); case kTfLiteBuiltinSplit: return VisitSplitOperator(delegateData, tfLiteContext, |