From 0176fd81b3f6a82ddc89e016cb634010f5397425 Mon Sep 17 00:00:00 2001
From: Keith Davis <keith.davis@arm.com>
Date: Tue, 1 Jun 2021 17:36:32 +0100
Subject: 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
---
 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 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,
-- 
cgit v1.2.1