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/armnnSerializer/Serializer.hpp | 3 +++ 1 file changed, 3 insertions(+) (limited to 'src/armnnSerializer/Serializer.hpp') diff --git a/src/armnnSerializer/Serializer.hpp b/src/armnnSerializer/Serializer.hpp index e4104dda8e..babecdc056 100644 --- a/src/armnnSerializer/Serializer.hpp +++ b/src/armnnSerializer/Serializer.hpp @@ -219,6 +219,9 @@ public: const armnn::QuantizedLstmInputParams& params, const char* name = nullptr) override; + void VisitRankLayer(const armnn::IConnectableLayer* layer, + const char* name = nullptr) override; + void VisitReshapeLayer(const armnn::IConnectableLayer* layer, const armnn::ReshapeDescriptor& reshapeDescriptor, const char* name = nullptr) override; -- cgit v1.2.1