diff options
| author | Matthew Bentham <matthew.bentham@arm.com> | 2019-05-13 10:02:45 +0100 |
|---|---|---|
| committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-05-13 16:41:09 +0000 |
| commit | acad04e3cb6abcdd9a3fcf4584db1cbedb52cb47 (patch) | |
| tree | 3e1497034566686e0648b1b17cbd0e8f72e91a94 /src/armnnSerializer/Serializer.cpp | |
| parent | 49cb7d0568b7d4c81889a1791a4ca4e6bd5c8f32 (diff) | |
| download | armnn-acad04e3cb6abcdd9a3fcf4584db1cbedb52cb47.tar.gz | |
MLCE-101 Add dilation parameter to serializer
Change-Id: I8142e179d38c7a2a9163cf3d30bd1f411e8e109c
Signed-off-by: Matthew Bentham <matthew.bentham@arm.com>
Diffstat (limited to 'src/armnnSerializer/Serializer.cpp')
| -rw-r--r-- | src/armnnSerializer/Serializer.cpp | 4 |
1 files changed, 4 insertions, 0 deletions
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)); |