commit acad04e3cb6abcdd9a3fcf4584db1cbedb52cb47
author Matthew Bentham <matthew.bentham@arm.com> Mon May 13 10:02:45 2019 +0100
committer Matteo Martincigh <matteo.martincigh@arm.com> Mon May 13 16:41:09 2019 +0000
tree 3e1497034566686e0648b1b17cbd0e8f72e91a94
parent 49cb7d0568b7d4c81889a1791a4ca4e6bd5c8f32

MLCE-101 Add dilation parameter to serializer

Change-Id: I8142e179d38c7a2a9163cf3d30bd1f411e8e109c
Signed-off-by: Matthew Bentham <matthew.bentham@arm.com>

src/armnnDeserializer/Deserializer.cpp

diff --git a/src/armnnDeserializer/Deserializer.cpp b/src/armnnDeserializer/Deserializer.cpp
index cbded60..8b790f7 100644
--- a/src/armnnDeserializer/Deserializer.cpp
+++ b/src/armnnDeserializer/Deserializer.cpp
@@ -980,6 +980,8 @@
     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());
 

src/armnnSerializer/ArmnnSchema.fbs

diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index e8d72fc..0419c4b 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -172,6 +172,8 @@
     padBottom:uint;
     strideX:uint;
     strideY:uint;
+    dilationX:uint = 1;
+    dilationY:uint = 1;
     biasEnabled:bool = false;
     dataLayout:DataLayout = NCHW;
 }
@@ -296,6 +298,8 @@
     padBottom:uint;
     strideX:uint;
     strideY:uint;
+    dilationX:uint = 1;
+    dilationY:uint = 1;
     biasEnabled:bool = false;
     dataLayout:DataLayout = NCHW;
 }

src/armnnSerializer/Serializer.cpp

diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp
index 0b8ad06..865ed7a 100644
--- a/src/armnnSerializer/Serializer.cpp
+++ b/src/armnnSerializer/Serializer.cpp
@@ -237,6 +237,8 @@
                                                               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 @@
                                                                descriptor.m_PadBottom,
                                                                descriptor.m_StrideX,
                                                                descriptor.m_StrideY,
+                                                               descriptor.m_DilationX,
+                                                               descriptor.m_DilationY,
                                                                descriptor.m_BiasEnabled,
                                                                GetFlatBufferDataLayout(descriptor.m_DataLayout));