From 74be13e0d663b9127f630360fffa55993ccce9d6 Mon Sep 17 00:00:00 2001 From: Matthew Sloyan Date: Wed, 3 May 2023 17:34:00 +0100 Subject: IVGCVSW-7618 Implement UnidirectionalSequenceLstm operator for Opaque Delegate * Intermediate tensors aren't accessible through the new Opaque interface yet, so we have to cast to TfLiteNode for now. Signed-off-by: Matthew Sloyan Change-Id: Ifd91131e5d5ff6cc057b80729fea9afa68ed240b --- delegate/opaque/src/armnn_delegate.cpp | 6 ++++++ 1 file changed, 6 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 cae1ea502b..fa64679efc 100644 --- a/delegate/opaque/src/armnn_delegate.cpp +++ b/delegate/opaque/src/armnn_delegate.cpp @@ -1131,6 +1131,12 @@ TfLiteStatus ArmnnSubgraph::VisitNode(DelegateData& delegateData, tfLiteNode, nodeIndex, kTfLiteBuiltinTransposeConv); + case kTfLiteBuiltinUnidirectionalSequenceLstm: + return VisitUnidirectionalSequenceLstmOperator(delegateData, + tfLiteContext, + tfLiteNode, + nodeIndex, + kTfLiteBuiltinUnidirectionalSequenceLstm); case kTfLiteBuiltinUnpack: return VisitUnpackOperator(delegateData, tfLiteContext, -- cgit v1.2.1