From 30b0020478652e441a5dff4880261f7c7007bb6c Mon Sep 17 00:00:00 2001 From: Nattapat Chaimanowong Date: Wed, 20 Feb 2019 17:31:34 +0000 Subject: IVGCVSW-2702 Add Serializer and Deserializer for Permute Change-Id: I56322cb935de42cfa25bd1fbb0c09d00c7a5dd2b Signed-off-by: Nattapat Chaimanowong --- src/armnnSerializer/Serializer.hpp | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'src/armnnSerializer/Serializer.hpp') diff --git a/src/armnnSerializer/Serializer.hpp b/src/armnnSerializer/Serializer.hpp index 0c442e0788..329b005624 100644 --- a/src/armnnSerializer/Serializer.hpp +++ b/src/armnnSerializer/Serializer.hpp @@ -72,6 +72,10 @@ public: armnn::LayerBindingId id, const char* name = nullptr) override; + void VisitPermuteLayer(const armnn::IConnectableLayer* layer, + const armnn::PermuteDescriptor& PermuteDescriptor, + const char* name = nullptr) override; + void VisitPooling2dLayer(const armnn::IConnectableLayer* layer, const armnn::Pooling2dDescriptor& pooling2dDescriptor, const char* name = nullptr) override; -- cgit v1.2.1