commit 4b227bb4e2d83f0e3125a2a8fcc6834b3b98b44d
author Sadik Armagan <sadik.armagan@arm.com> Fri Jan 22 10:53:38 2021 +0000
committer Sadik Armagan <sadik.armagan@arm.com> Tue Jan 26 09:40:33 2021 +0000
tree 2ec6f1063008929df60de043ca6c882b9a847e4c
parent 0b51d5ad533f8ecde71f957077690195eea29ffc

IVGCVSW-5390 'TfLiteDelegate: Implement the Normalisation operators'

* Added L2_NORMALIZATION and LOCAL_RESPONSE_NORMALIZATION opertor support

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

delegate/src/armnn_delegate.cpp

diff --git a/delegate/src/armnn_delegate.cpp b/delegate/src/armnn_delegate.cpp
index 6dba890..6f72d86 100644
--- a/delegate/src/armnn_delegate.cpp
+++ b/delegate/src/armnn_delegate.cpp
@@ -575,11 +575,11 @@
                                            nodeIndex,
                                            kTfLiteBuiltinHardSwish);
         case kTfLiteBuiltinL2Normalization:
-            return VisitNormalizationOperator(delegateData,
-                                              tfLiteContext,
-                                              tfLiteNode,
-                                              nodeIndex,
-                                              kTfLiteBuiltinL2Normalization);
+            return VisitL2NormalizationOperator(delegateData,
+                                                tfLiteContext,
+                                                tfLiteNode,
+                                                nodeIndex,
+                                                kTfLiteBuiltinL2Normalization);
         case kTfLiteBuiltinL2Pool2d:
             return VisitPoolingOperator(delegateData,
                                         tfLiteContext,
@@ -599,11 +599,11 @@
                                            nodeIndex,
                                            kTfLiteBuiltinLessEqual);
         case kTfLiteBuiltinLocalResponseNormalization:
-            return VisitNormalizationOperator(delegateData,
-                                              tfLiteContext,
-                                              tfLiteNode,
-                                              nodeIndex,
-                                              kTfLiteBuiltinLocalResponseNormalization);
+            return VisitLocalResponseNormalizationOperator(delegateData,
+                                                           tfLiteContext,
+                                                           tfLiteNode,
+                                                           nodeIndex,
+                                                           kTfLiteBuiltinLocalResponseNormalization);
         case kTfLiteBuiltinLogicalAnd:
             return VisitLogicalBinaryOperator(delegateData,
                                               tfLiteContext,