From b2831c5d179c5853c884280107d2420904441a2c Mon Sep 17 00:00:00 2001 From: Kevin May Date: Wed, 26 Apr 2023 17:27:24 +0100 Subject: IVGCVSW-7591 IVGCVSW-7592 Add Gather and GatherNd to Opaque Delegate Signed-off-by: Kevin May Change-Id: Id2b6a4f70b1cb50e5f7f7ab4e30487b3816c9ad4 --- delegate/opaque/src/armnn_delegate.cpp | 12 ++++++++++++ 1 file changed, 12 insertions(+) (limited to 'delegate/opaque/src/armnn_delegate.cpp') 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, -- cgit v1.2.1