diff options
| author | Sadik Armagan <sadik.armagan@arm.com> | 2021-01-22 10:53:38 +0000 |
|---|---|---|
| committer | Sadik Armagan <sadik.armagan@arm.com> | 2021-01-26 09:40:33 +0000 |
| commit | 4b227bb4e2d83f0e3125a2a8fcc6834b3b98b44d (patch) | |
| tree | 2ec6f1063008929df60de043ca6c882b9a847e4c /delegate/src/armnn_delegate.cpp | |
| parent | 0b51d5ad533f8ecde71f957077690195eea29ffc (diff) | |
| download | armnn-4b227bb4e2d83f0e3125a2a8fcc6834b3b98b44d.tar.gz | |
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
Diffstat (limited to 'delegate/src/armnn_delegate.cpp')
| -rw-r--r-- | delegate/src/armnn_delegate.cpp | 20 |
1 files changed, 10 insertions, 10 deletions
diff --git a/delegate/src/armnn_delegate.cpp b/delegate/src/armnn_delegate.cpp index 6dba890509..6f72d864b9 100644 --- a/delegate/src/armnn_delegate.cpp +++ b/delegate/src/armnn_delegate.cpp @@ -575,11 +575,11 @@ TfLiteStatus ArmnnSubgraph::VisitNode(DelegateData& delegateData, 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 @@ TfLiteStatus ArmnnSubgraph::VisitNode(DelegateData& delegateData, nodeIndex, kTfLiteBuiltinLessEqual); case kTfLiteBuiltinLocalResponseNormalization: - return VisitNormalizationOperator(delegateData, - tfLiteContext, - tfLiteNode, - nodeIndex, - kTfLiteBuiltinLocalResponseNormalization); + return VisitLocalResponseNormalizationOperator(delegateData, + tfLiteContext, + tfLiteNode, + nodeIndex, + kTfLiteBuiltinLocalResponseNormalization); case kTfLiteBuiltinLogicalAnd: return VisitLogicalBinaryOperator(delegateData, tfLiteContext, |