diff options
Diffstat (limited to 'src/armnnSerializer')
-rw-r--r-- | src/armnnSerializer/ArmnnSchema.fbs | 4 | ||||
-rw-r--r-- | src/armnnSerializer/Serializer.cpp | 4 |
2 files changed, 8 insertions, 0 deletions
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)); |