diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-06-21 14:20:11 +0100 |
---|---|---|
committer | Áron Virginás-Tar <aron.virginas-tar@arm.com> | 2019-06-21 14:30:24 +0000 |
commit | 5e1b0cf8a7519afb49874a83429ef9939a249f0d (patch) | |
tree | 5c5153572cbc5ddba779af24f0bb81350e0ec4d5 /src/armnnDeserializer | |
parent | cb549301bc4c5a405e02c1f433537557423d2e36 (diff) | |
download | armnn-5e1b0cf8a7519afb49874a83429ef9939a249f0d.tar.gz |
IVGCVSW-3330 Parse dilation parameters for DepthwiseConvolution2d in Deserializer
* Added code for reading parameters dilationX and dilationY
to Deserializer::ParseDepthwiseConvolution()
* Updated serialization unit tests to check for dilation params
for both Convolution2d and DepthwiseConvolution2d
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: Id03629866009e799cf7daf8b7bdafc73d158d9bb
Diffstat (limited to 'src/armnnDeserializer')
-rw-r--r-- | src/armnnDeserializer/Deserializer.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/armnnDeserializer/Deserializer.cpp b/src/armnnDeserializer/Deserializer.cpp index 0569d8b000..452ef8e5ae 100644 --- a/src/armnnDeserializer/Deserializer.cpp +++ b/src/armnnDeserializer/Deserializer.cpp @@ -1046,7 +1046,9 @@ void Deserializer::ParseDepthwiseConvolution2d(GraphPtr graph, unsigned int laye descriptor.m_PadTop = serializerDescriptor->padTop(); descriptor.m_PadBottom = serializerDescriptor->padBottom(); descriptor.m_StrideX = serializerDescriptor->strideX(); - descriptor.m_StrideY = serializerDescriptor->strideY();; + 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()); |