commit 81b66f3aeea1d0e788b0ce2894a58fedc763470b
author Kevin May <kevin.may@arm.com> Wed Apr 26 14:55:36 2023 +0100
committer Kevin May <kevin.may@arm.com> Wed Apr 26 16:24:25 2023 +0000
tree b27461cfb84b38b52bc58e63fdd83d472aac3af2
parent 024ef0b460c802a7c841dcba4b7e894e714d4512

IVGCVSW-7577, IVGCVSW-7578 Implement BatchToSpaceNd and SpaceToBatchNd in Opaque Delegate

Signed-off-by: Kevin May <kevin.may@arm.com>
Change-Id: I38304abce1a417bb69aced2a5b38e976ea0cbbc0

delegate/opaque/src/armnn_delegate.cpp

diff --git a/delegate/opaque/src/armnn_delegate.cpp b/delegate/opaque/src/armnn_delegate.cpp
index c305c40..7f3d8cf 100644
--- a/delegate/opaque/src/armnn_delegate.cpp
+++ b/delegate/opaque/src/armnn_delegate.cpp
@@ -622,6 +622,12 @@
 {
     switch (TfLiteRegistrationExternalGetBuiltInCode(tfLiteRegistration))
     {
+        case kTfLiteBuiltinBatchToSpaceNd:
+            return VisitBatchToSpaceNdOperator(delegateData,
+                                               tfLiteContext,
+                                               tfLiteNode,
+                                               nodeIndex,
+                                               kTfLiteBuiltinBatchToSpaceNd);
         case kTfLiteBuiltinCast:
             return VisitCastOperator(delegateData,
                                      tfLiteContext,
@@ -688,6 +694,12 @@
                                            tfLiteNode,
                                            nodeIndex,
                                            kTfLiteBuiltinNotEqual);
+        case kTfLiteBuiltinSpaceToBatchNd:
+            return VisitSpaceToBatchNdOperator(delegateData,
+                                               tfLiteContext,
+                                               tfLiteNode,
+                                               nodeIndex,
+                                               kTfLiteBuiltinSpaceToBatchNd);
         default:
             return kTfLiteError;
     }