From 8ed39ae450a077c7e4d672b5f05ff1d68ee67aab Mon Sep 17 00:00:00 2001 From: Narumol Prangnawarat Date: Thu, 15 Jul 2021 16:16:25 +0100 Subject: MLCE-530 Add front end support for UnidirectionalSequenceLstm on ArmNN Signed-off-by: Narumol Prangnawarat Change-Id: I57bcbdec3eb0155f41af0fe7d6abf9bac2ec86eb --- src/armnn/LayersFwd.hpp | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'src/armnn/LayersFwd.hpp') diff --git a/src/armnn/LayersFwd.hpp b/src/armnn/LayersFwd.hpp index cdbcaa7e90..e3ae23cf40 100644 --- a/src/armnn/LayersFwd.hpp +++ b/src/armnn/LayersFwd.hpp @@ -73,6 +73,7 @@ #include "layers/SwitchLayer.hpp" #include "layers/TransposeConvolution2dLayer.hpp" #include "layers/TransposeLayer.hpp" +#include "layers/UnidirectionalSequenceLstmLayer.hpp" #include "layers/UnmapLayer.hpp" namespace armnn @@ -107,6 +108,7 @@ DECLARE_LAYER(Addition) DECLARE_LAYER(ArgMinMax) DECLARE_LAYER(BatchNormalization) DECLARE_LAYER(BatchToSpaceNd) +DECLARE_LAYER(Cast) DECLARE_LAYER(Comparison) DECLARE_LAYER(Concat) DECLARE_LAYER(Constant) @@ -168,6 +170,6 @@ DECLARE_LAYER(Subtraction) DECLARE_LAYER(Switch) DECLARE_LAYER(Transpose) DECLARE_LAYER(TransposeConvolution2d) +DECLARE_LAYER(UnidirectionalSequenceLstm) DECLARE_LAYER(Unmap) -DECLARE_LAYER(Cast) } -- cgit v1.2.1