From 6b4ed983aec66a979fbf206291d387bad0eeec9d Mon Sep 17 00:00:00 2001 From: Nattapat Chaimanowong Date: Tue, 26 Feb 2019 17:24:13 +0000 Subject: IVGCVSW-2682 Add Serializer and Deserializer for BatchToSpaceNd Change-Id: I8b8ac99b4ecf282b796a8af4b30594ade4e061e6 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 a86e04f05d..f8e2b16bf9 100644 --- a/src/armnnSerializer/Serializer.hpp +++ b/src/armnnSerializer/Serializer.hpp @@ -49,6 +49,10 @@ public: void VisitAdditionLayer(const armnn::IConnectableLayer* layer, const char* name = nullptr) override; + void VisitBatchToSpaceNdLayer(const armnn::IConnectableLayer* layer, + const armnn::BatchToSpaceNdDescriptor& descriptor, + const char* name = nullptr) override; + void VisitConstantLayer(const armnn::IConnectableLayer* layer, const armnn::ConstTensor& input, const char* = nullptr) override; -- cgit v1.2.1