diff options
| author | Kevin May <kevin.may@arm.com> | 2023-04-26 17:27:24 +0100 |
|---|---|---|
| committer | Kevin May <kevin.may@arm.com> | 2023-04-27 08:43:00 +0000 |
| commit | b2831c5d179c5853c884280107d2420904441a2c (patch) | |
| tree | ca468711da2c5032f7c308245f1beac9286b7863 /delegate/opaque/src/armnn_delegate.cpp | |
| parent | 559d9097ac8e95d53d507074dbb1ba69d42664a6 (diff) | |
| download | armnn-b2831c5d179c5853c884280107d2420904441a2c.tar.gz | |
IVGCVSW-7591 IVGCVSW-7592 Add Gather and GatherNd to Opaque Delegate
Signed-off-by: Kevin May <kevin.may@arm.com>
Change-Id: Id2b6a4f70b1cb50e5f7f7ab4e30487b3816c9ad4
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 4d43b9271e..38a67e7c46 100644 --- a/delegate/opaque/src/armnn_delegate.cpp +++ b/delegate/opaque/src/armnn_delegate.cpp @@ -670,6 +670,18 @@ TfLiteStatus ArmnnSubgraph::VisitNode(DelegateData& delegateData, tfLiteNode, nodeIndex, kTfLiteBuiltinEqual); + case kTfLiteBuiltinGather: + return VisitGatherOperator(delegateData, + tfLiteContext, + tfLiteNode, + nodeIndex, + kTfLiteBuiltinGather); + case kTfLiteBuiltinGatherNd: + return VisitGatherNdOperator(delegateData, + tfLiteContext, + tfLiteNode, + nodeIndex, + kTfLiteBuiltinGatherNd); case kTfLiteBuiltinGreater: return VisitComparisonOperator(delegateData, tfLiteContext, |