commit 937565b90bb33eea785898c44db2942dd7af56e7
author Sadik Armagan <sadik.armagan@arm.com> Wed Apr 21 14:03:28 2021 +0100
committer Sadik Armagan <sadik.armagan@arm.com> Fri Apr 23 06:55:49 2021 +0000
tree c033c6a364a24475e7f8e723519b26ddfddaedc3
parent 48f011e1d5337ff580ae2c315f670bfc15413598

IVGCVSW-5430 'Add CAST Operator Support to Delegate'

Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I6da711950b8e7d3c0d5cbd443e91eb36700ac4c8

delegate/src/armnn_delegate.cpp

diff --git a/delegate/src/armnn_delegate.cpp b/delegate/src/armnn_delegate.cpp
index 2b07fc7..4c625a9 100644
--- a/delegate/src/armnn_delegate.cpp
+++ b/delegate/src/armnn_delegate.cpp
@@ -486,6 +486,12 @@
                                                tfLiteNode,
                                                nodeIndex,
                                                kTfLiteBuiltinBatchToSpaceNd);
+        case kTfLiteBuiltinCast:
+            return VisitCastOperator(delegateData,
+                                     tfLiteContext,
+                                     tfLiteNode,
+                                     nodeIndex,
+                                     kTfLiteBuiltinCast);
         case kTfLiteBuiltinConcatenation:
             return VisitControlOperator(delegateData,
                                         tfLiteContext,