diff options
author | Francis Murtagh <francis.murtagh@arm.com> | 2023-04-26 15:58:39 +0100 |
---|---|---|
committer | Francis Murtagh <francis.murtagh@arm.com> | 2023-04-27 17:27:09 +0100 |
commit | 3a9e7bac89a120fddaf3bcf312105258a840dab1 (patch) | |
tree | 381db8bb43e9fdfe822bdfe2bf44395cf8296dd1 /delegate/opaque/src/armnn_delegate.cpp | |
parent | cfbb2717b21fec803839c478dde8292263fa4f45 (diff) | |
download | armnn-3a9e7bac89a120fddaf3bcf312105258a840dab1.tar.gz |
IVGCVSW-7584 IVGCVSW-7585 Implement Conv3d and TransposeConv2d operators for opaque delegate
* Add check for TF version so tests work with opaque and classic
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>
Change-Id: I3a6699150afabfc6200e1cd4d264a1f7c65c5f60
Diffstat (limited to 'delegate/opaque/src/armnn_delegate.cpp')
-rw-r--r-- | delegate/opaque/src/armnn_delegate.cpp | 14 |
1 files changed, 13 insertions, 1 deletions
diff --git a/delegate/opaque/src/armnn_delegate.cpp b/delegate/opaque/src/armnn_delegate.cpp index d631d919b6..bffbca8111 100644 --- a/delegate/opaque/src/armnn_delegate.cpp +++ b/delegate/opaque/src/armnn_delegate.cpp @@ -672,6 +672,12 @@ TfLiteStatus ArmnnSubgraph::VisitNode(DelegateData& delegateData, tfLiteNode, nodeIndex, kTfLiteBuiltinConv2d); + case kTfLiteBuiltinConv3d: + return VisitConvolutionOperator(delegateData, + tfLiteContext, + tfLiteNode, + nodeIndex, + kTfLiteBuiltinConv3d); case kTfLiteBuiltinDepthwiseConv2d: return VisitConvolutionOperator(delegateData, tfLiteContext, @@ -873,8 +879,14 @@ TfLiteStatus ArmnnSubgraph::VisitNode(DelegateData& delegateData, tfLiteNode, nodeIndex, kTfLiteBuiltinTanh); + case kTfLiteBuiltinTransposeConv: + return VisitConvolutionOperator(delegateData, + tfLiteContext, + tfLiteNode, + nodeIndex, + kTfLiteBuiltinTransposeConv); default: return kTfLiteError; } } -} // armnnOpaqueDelegate namespace
\ No newline at end of file +} // armnnOpaqueDelegate namespace |