From acad04e3cb6abcdd9a3fcf4584db1cbedb52cb47 Mon Sep 17 00:00:00 2001 From: Matthew Bentham Date: Mon, 13 May 2019 10:02:45 +0100 Subject: MLCE-101 Add dilation parameter to serializer Change-Id: I8142e179d38c7a2a9163cf3d30bd1f411e8e109c Signed-off-by: Matthew Bentham --- src/armnnDeserializer/Deserializer.cpp | 2 ++ src/armnnSerializer/ArmnnSchema.fbs | 4 ++++ src/armnnSerializer/Serializer.cpp | 4 ++++ 3 files changed, 10 insertions(+) diff --git a/src/armnnDeserializer/Deserializer.cpp b/src/armnnDeserializer/Deserializer.cpp index cbded60f93..8b790f7b04 100644 --- a/src/armnnDeserializer/Deserializer.cpp +++ b/src/armnnDeserializer/Deserializer.cpp @@ -980,6 +980,8 @@ void Deserializer::ParseConvolution2d(GraphPtr graph, unsigned int layerIndex) descriptor.m_PadBottom = serializerDescriptor->padBottom(); descriptor.m_StrideX = serializerDescriptor->strideX(); descriptor.m_StrideY = serializerDescriptor->strideY();; + descriptor.m_DilationX = serializerDescriptor->dilationX(); + descriptor.m_DilationY = serializerDescriptor->dilationY();; descriptor.m_BiasEnabled = serializerDescriptor->biasEnabled();; descriptor.m_DataLayout = ToDataLayout(serializerDescriptor->dataLayout()); diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs index e8d72fc997..0419c4b883 100644 --- a/src/armnnSerializer/ArmnnSchema.fbs +++ b/src/armnnSerializer/ArmnnSchema.fbs @@ -172,6 +172,8 @@ table Convolution2dDescriptor { padBottom:uint; strideX:uint; strideY:uint; + dilationX:uint = 1; + dilationY:uint = 1; biasEnabled:bool = false; dataLayout:DataLayout = NCHW; } @@ -296,6 +298,8 @@ table DepthwiseConvolution2dDescriptor { padBottom:uint; strideX:uint; strideY:uint; + dilationX:uint = 1; + dilationY:uint = 1; biasEnabled:bool = false; dataLayout:DataLayout = NCHW; } diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp index 0b8ad06d18..865ed7af51 100644 --- a/src/armnnSerializer/Serializer.cpp +++ b/src/armnnSerializer/Serializer.cpp @@ -237,6 +237,8 @@ void SerializerVisitor::VisitConvolution2dLayer(const armnn::IConnectableLayer* descriptor.m_PadBottom, descriptor.m_StrideX, descriptor.m_StrideY, + descriptor.m_DilationX, + descriptor.m_DilationY, descriptor.m_BiasEnabled, GetFlatBufferDataLayout(descriptor.m_DataLayout)); auto flatBufferWeightsConstTensorInfo = CreateConstTensorInfo(weights); @@ -272,6 +274,8 @@ void SerializerVisitor::VisitDepthwiseConvolution2dLayer(const armnn::IConnectab descriptor.m_PadBottom, descriptor.m_StrideX, descriptor.m_StrideY, + descriptor.m_DilationX, + descriptor.m_DilationY, descriptor.m_BiasEnabled, GetFlatBufferDataLayout(descriptor.m_DataLayout)); -- cgit v1.2.1