diff options
Diffstat (limited to 'src/armnn/Network.hpp')
-rw-r--r-- | src/armnn/Network.hpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/armnn/Network.hpp b/src/armnn/Network.hpp index 8a99debb47..679ab51d43 100644 --- a/src/armnn/Network.hpp +++ b/src/armnn/Network.hpp @@ -6,6 +6,7 @@ #include <armnn/DescriptorsFwd.hpp> #include <armnn/LstmParams.hpp> +#include <armnn/QuantizedLstmParams.hpp> #include <armnn/TensorFwd.hpp> #include <armnn/Types.hpp> @@ -200,6 +201,9 @@ public: IConnectableLayer* AddStackLayer(const StackDescriptor& stackDescriptor, const char* name = nullptr) override; + IConnectableLayer* AddQuantizedLstmLayer(const QuantizedLstmInputParams& params, + const char* name = nullptr) override; + void Accept(ILayerVisitor& visitor) const override; private: |