From 22972f04d6aa5c4d1269ed3be8bc5fb7b508dcda Mon Sep 17 00:00:00 2001 From: Bruno Goncalves Date: Fri, 26 Apr 2019 21:03:24 -0300 Subject: MLCE-101 Add dilation support for DepthWiseConv workload Adds unit tests for dilated depthwise conv Change-Id: Iad0a1b33d07fb0ef8f9f6edf0fd0f83a5800a36d Signed-off-by: Bruno Goncalves Signed-off-by: Matthew Bentham --- src/backends/backendsCommon/test/Conv2dTestImpl.hpp | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) (limited to 'src/backends/backendsCommon/test/Conv2dTestImpl.hpp') diff --git a/src/backends/backendsCommon/test/Conv2dTestImpl.hpp b/src/backends/backendsCommon/test/Conv2dTestImpl.hpp index c2e539b20f..bb5656bd01 100755 --- a/src/backends/backendsCommon/test/Conv2dTestImpl.hpp +++ b/src/backends/backendsCommon/test/Conv2dTestImpl.hpp @@ -827,7 +827,9 @@ LayerTestResult DepthwiseConvolution2dNhwcTestImpl( uint32_t padRight = 0, uint32_t padBottom = 0, uint32_t strideX = 1, - uint32_t strideY = 1) + uint32_t strideY = 1, + uint32_t dilationX = 1, + uint32_t dilationY = 1) { unsigned int inputNum = boost::numeric_cast(input.shape()[0]); unsigned int inputChannels = boost::numeric_cast(input.shape()[3]); @@ -894,6 +896,8 @@ LayerTestResult DepthwiseConvolution2dNhwcTestImpl( data.m_Parameters.m_PadTop = padTop; data.m_Parameters.m_PadBottom = padBottom; data.m_Parameters.m_DataLayout = armnn::DataLayout::NHWC; + data.m_Parameters.m_DilationX = dilationX; + data.m_Parameters.m_DilationY = dilationY; armnn::WorkloadInfo info; AddInputToWorkload(data, info, inputTensorInfo, inputHandle.get()); -- cgit v1.2.1