diff options
| author | Francis Murtagh <francis.murtagh@arm.com> | 2023-04-28 14:05:43 +0100 |
|---|---|---|
| committer | TeresaARM <teresa.charlinreyes@arm.com> | 2023-04-28 15:44:11 +0000 |
| commit | 36d94ef824c516f27fad8f17a96e2123565e6f9f (patch) | |
| tree | abe5852f521f1b6324b01fddab8bfeaa2e645b60 /delegate/opaque/src/armnn_delegate.cpp | |
| parent | ecebb0f321d996a276bfe79efd9c8bef25834a92 (diff) | |
| download | armnn-36d94ef824c516f27fad8f17a96e2123565e6f9f.tar.gz | |
IVGCVSW-7602 IVGCVSW-7602 Implement opaque delegate for Quantize + Dequantize operator
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>
Change-Id: I318cb25f526dfe0f7aa6afcf77971afd8d437209
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 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, |