aboutsummaryrefslogtreecommitdiff
path: root/include/armnn/INetwork.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'include/armnn/INetwork.hpp')
-rw-r--r--include/armnn/INetwork.hpp9
1 files changed, 9 insertions, 0 deletions
diff --git a/include/armnn/INetwork.hpp b/include/armnn/INetwork.hpp
index b40db62a59..865d1291a9 100644
--- a/include/armnn/INetwork.hpp
+++ b/include/armnn/INetwork.hpp
@@ -691,6 +691,15 @@ public:
IConnectableLayer* AddLogicalBinaryLayer(const LogicalBinaryDescriptor& descriptor,
const char* name = nullptr);
+ /// Add a UnidirectionalSequenceLstm layer to the network
+ /// @param descriptor - Parameters for the UnidirectionalSequenceLstm operation
+ /// @param params - Weights and biases for the UnidirectionalSequenceLstm
+ /// @param name - Optional name for the layer
+ /// @return - Interface for configuring the layer.
+ IConnectableLayer* AddUnidirectionalSequenceLstmLayer(const UnidirectionalSequenceLstmDescriptor& descriptor,
+ const LstmInputParams& params,
+ const char* name = nullptr);
+
void Accept(ILayerVisitor& visitor) const;
void ExecuteStrategy(IStrategy& strategy) const;