From 36d94ef824c516f27fad8f17a96e2123565e6f9f Mon Sep 17 00:00:00 2001 From: Francis Murtagh Date: Fri, 28 Apr 2023 14:05:43 +0100 Subject: IVGCVSW-7602 IVGCVSW-7602 Implement opaque delegate for Quantize + Dequantize operator Signed-off-by: Francis Murtagh Change-Id: I318cb25f526dfe0f7aa6afcf77971afd8d437209 --- 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 b38b528317..8c3ddfaeea 100644 --- a/delegate/opaque/src/armnn_delegate.cpp +++ b/delegate/opaque/src/armnn_delegate.cpp @@ -726,6 +726,12 @@ TfLiteStatus ArmnnSubgraph::VisitNode(DelegateData& delegateData, tfLiteNode, nodeIndex, kTfLiteBuiltinDepthwiseConv2d); + case kTfLiteBuiltinDequantize: + return VisitDequantizeOperator(delegateData, + tfLiteContext, + tfLiteNode, + nodeIndex, + kTfLiteBuiltinDequantize); case kTfLiteBuiltinDiv: return VisitElementwiseBinaryOperator(delegateData, tfLiteContext, @@ -942,6 +948,12 @@ TfLiteStatus ArmnnSubgraph::VisitNode(DelegateData& delegateData, tfLiteNode, nodeIndex, kTfLiteBuiltinPrelu); + case kTfLiteBuiltinQuantize: + return VisitQuantizeOperator(delegateData, + tfLiteContext, + tfLiteNode, + nodeIndex, + kTfLiteBuiltinQuantize); case kTfLiteBuiltinRelu: return VisitActivationOperator(delegateData, tfLiteContext, -- cgit v1.2.1