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
diff --git a/delegate/opaque/src/armnn_delegate.cpp b/delegate/opaque/src/armnn_delegate.cpp
index d631d91..bffbca8 100644
--- a/delegate/opaque/src/armnn_delegate.cpp
+++ b/delegate/opaque/src/armnn_delegate.cpp
@@ -672,6 +672,12 @@
tfLiteNode,
nodeIndex,
kTfLiteBuiltinConv2d);
+ case kTfLiteBuiltinConv3d:
+ return VisitConvolutionOperator(delegateData,
+ tfLiteContext,
+ tfLiteNode,
+ nodeIndex,
+ kTfLiteBuiltinConv3d);
case kTfLiteBuiltinDepthwiseConv2d:
return VisitConvolutionOperator(delegateData,
tfLiteContext,
@@ -873,8 +879,14 @@
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