diff options
| author | John Mcloughlin <john.mcloughlin@arm.com> | 2023-04-26 20:14:47 +0100 |
|---|---|---|
| committer | John Mcloughlin <john.mcloughlin@arm.com> | 2023-04-26 20:14:47 +0100 |
| commit | 559d9097ac8e95d53d507074dbb1ba69d42664a6 (patch) | |
| tree | 30de17c6ff453764758d05f7ff6e5b09817b89f6 /delegate/opaque/src/armnn_delegate.cpp | |
| parent | 81b66f3aeea1d0e788b0ce2894a58fedc763470b (diff) | |
| download | armnn-559d9097ac8e95d53d507074dbb1ba69d42664a6.tar.gz | |
IVGCVSW-7575 Implement opaque delegate for ArgMinMax operator
* Added Opaque ArgMinMax and associated test cases
Signed-off-by: John Mcloughlin <john.mcloughlin@arm.com>
Change-Id: I098b94cc35707370a0bbc7456bfdd48bb47432f0
Diffstat (limited to 'delegate/opaque/src/armnn_delegate.cpp')
| -rw-r--r-- | delegate/opaque/src/armnn_delegate.cpp | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/delegate/opaque/src/armnn_delegate.cpp b/delegate/opaque/src/armnn_delegate.cpp index 7f3d8cf9e9..4d43b9271e 100644 --- a/delegate/opaque/src/armnn_delegate.cpp +++ b/delegate/opaque/src/armnn_delegate.cpp @@ -622,6 +622,18 @@ TfLiteStatus ArmnnSubgraph::VisitNode(DelegateData& delegateData, { switch (TfLiteRegistrationExternalGetBuiltInCode(tfLiteRegistration)) { + case kTfLiteBuiltinArgMax: + return VisitArgMinMaxOperator(delegateData, + tfLiteContext, + tfLiteNode, + nodeIndex, + kTfLiteBuiltinArgMax); + case kTfLiteBuiltinArgMin: + return VisitArgMinMaxOperator(delegateData, + tfLiteContext, + tfLiteNode, + nodeIndex, + kTfLiteBuiltinArgMin); case kTfLiteBuiltinBatchToSpaceNd: return VisitBatchToSpaceNdOperator(delegateData, tfLiteContext, |