From 2605b236d103e1ba27069e0d668599042a4761af Mon Sep 17 00:00:00 2001 From: Finn Williams Date: Wed, 10 Jun 2020 15:53:46 +0100 Subject: IVGCVSW-4624 Add a RANK Reference Implementation * Add Rank front end * Add Rank reference implementation * Add Rank serialization support * Add Scalar serialization support Signed-off-by: Finn Williams Change-Id: I06e4a468c2a84e79bae2e6c5348596bbbf853b4b --- src/armnn/LayersFwd.hpp | 2 ++ 1 file changed, 2 insertions(+) (limited to 'src/armnn/LayersFwd.hpp') diff --git a/src/armnn/LayersFwd.hpp b/src/armnn/LayersFwd.hpp index 575c3e5c68..fda00df7cc 100644 --- a/src/armnn/LayersFwd.hpp +++ b/src/armnn/LayersFwd.hpp @@ -53,6 +53,7 @@ #include "layers/QuantizeLayer.hpp" #include "layers/QLstmLayer.hpp" #include "layers/QuantizedLstmLayer.hpp" +#include "layers/RankLayer.hpp" #include "layers/ReshapeLayer.hpp" #include "layers/ResizeLayer.hpp" #include "layers/SliceLayer.hpp" @@ -142,6 +143,7 @@ DECLARE_LAYER(Prelu) DECLARE_LAYER(Quantize) DECLARE_LAYER(QLstm) DECLARE_LAYER(QuantizedLstm) +DECLARE_LAYER(Rank) DECLARE_LAYER(Reshape) DECLARE_LAYER(Resize) DECLARE_LAYER(Slice) -- cgit v1.2.1