From b63a31170aee1d28267d83a4bc67b57708fb6b05 Mon Sep 17 00:00:00 2001 From: Matthew Sloyan Date: Wed, 8 Sep 2021 13:05:51 +0100 Subject: IVGCVSW-6163 Add Conv3d FrontEnd and Ref Implementation * Added front-end * Added Reference workload * Added Serializer & Deserializer support * Added unit tests * Added NDHWC DataLayout Signed-off-by: Matthew Sloyan Change-Id: Iec4d39e7433b5334d52fa44cf8efc6bcd39319d8 --- src/armnnSerializer/Serializer.hpp | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) (limited to 'src/armnnSerializer/Serializer.hpp') diff --git a/src/armnnSerializer/Serializer.hpp b/src/armnnSerializer/Serializer.hpp index 43fb0f46b1..1161095c33 100644 --- a/src/armnnSerializer/Serializer.hpp +++ b/src/armnnSerializer/Serializer.hpp @@ -144,12 +144,17 @@ private: const char* name = nullptr); void SerializeConstantLayer(const armnn::IConnectableLayer* layer, - const std::vector& contants, + const std::vector& constants, const char* name = nullptr); void SerializeConvolution2dLayer(const armnn::IConnectableLayer* layer, const armnn::Convolution2dDescriptor& descriptor, - const std::vector& contants, + const std::vector& constants, + const char* name = nullptr); + + void SerializeConvolution3dLayer(const armnn::IConnectableLayer* layer, + const armnn::Convolution3dDescriptor& descriptor, + const std::vector& constants, const char* name = nullptr); void SerializeDepthToSpaceLayer(const armnn::IConnectableLayer* layer, -- cgit v1.2.1