commit 7684b18e8fec45355a49e7f7165c582efc553ab6
author Narumol Prangnawarat <narumol.prangnawarat@arm.com> Thu Aug 12 14:48:15 2021 +0100
committer Narumol Prangnawarat <narumol.prangnawarat@arm.com> Tue Aug 31 14:13:48 2021 +0000
tree f79078cec5e14534b82effaa6f8c4b70c3fe6bff
parent b338fb08cc77892f9cc19132421db06d77f9afde

MLCE-530 Add support for UnidirectionalSequenceLstm to armnn delegate

Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Ib04f8d6b9e60a4204c56eba4c2ecd2b316509dcc

delegate/src/armnn_delegate.cpp

diff --git a/delegate/src/armnn_delegate.cpp b/delegate/src/armnn_delegate.cpp
index 0ac3380..d3ccecb 100644
--- a/delegate/src/armnn_delegate.cpp
+++ b/delegate/src/armnn_delegate.cpp
@@ -36,6 +36,7 @@
 #include "SpaceDepth.hpp"
 #include "Split.hpp"
 #include "Transpose.hpp"
+#include "UnidirectionalSequenceLstm.hpp"
 #include "Unpack.hpp"
 
 #include <flatbuffers/flatbuffers.h>
@@ -890,6 +891,12 @@
                                            tfLiteNode,
                                            nodeIndex,
                                            kTfLiteBuiltinTanh);
+        case kTfLiteBuiltinUnidirectionalSequenceLstm:
+            return VisitUnidirectionalSequenceLstmOperator(delegateData,
+                                                           tfLiteContext,
+                                                           tfLiteNode,
+                                                           nodeIndex,
+                                                           kTfLiteBuiltinUnidirectionalSequenceLstm);
         case kTfLiteBuiltinUnpack:
             return VisitUnpackOperator(delegateData,
                                        tfLiteContext,