diff options
| author | David Monahan <david.monahan@arm.com> | 2023-05-03 15:53:03 +0100 |
|---|---|---|
| committer | David Monahan <david.monahan@arm.com> | 2023-05-04 12:37:04 +0000 |
| commit | c833cef6240abb941725a667042b84b936f1e86f (patch) | |
| tree | 3f6c984a4507f4d77bfa87954f54b9d76e1e2268 /delegate/opaque/src/armnn_delegate.cpp | |
| parent | 6aed53be8c5d1af50125a7829b5a13c760f2aee2 (diff) | |
| download | armnn-c833cef6240abb941725a667042b84b936f1e86f.tar.gz | |
IVGCVSW-7612 Implement Split Operators
* Added implementation for Split / SplitV in Opaque Delegate
Signed-off-by: David Monahan <david.monahan@arm.com>
Change-Id: I1ad2925aece7d4e0567e63ecb768348152a70a7a
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 2fd8142169..f7476d17c5 100644 --- a/delegate/opaque/src/armnn_delegate.cpp +++ b/delegate/opaque/src/armnn_delegate.cpp @@ -1064,6 +1064,18 @@ TfLiteStatus ArmnnSubgraph::VisitNode(DelegateData& delegateData, tfLiteNode, nodeIndex, kTfLiteBuiltinSpaceToDepth); + case kTfLiteBuiltinSplit: + return VisitSplitOperator(delegateData, + tfLiteContext, + tfLiteNode, + nodeIndex, + kTfLiteBuiltinSplit); + case kTfLiteBuiltinSplitV: + return VisitSplitVOperator(delegateData, + tfLiteContext, + tfLiteNode, + nodeIndex, + kTfLiteBuiltinSplitV); case kTfLiteBuiltinSub: return VisitElementwiseBinaryOperator(delegateData, tfLiteContext, |