diff options
| author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2019-06-14 11:09:19 +0100 |
|---|---|---|
| committer | Teresa Charlin <teresa.charlinreyes@arm.com> | 2019-06-18 15:20:43 +0100 |
| commit | edeeb168c6d2612d5b934158bf18c35c843dfc3e (patch) | |
| tree | 49aa615311c9906dd1bf20694eb6f7b45cfc20fd /src/backends/backendsCommon/test/Conv2dTestImpl.hpp | |
| parent | 0e406eed386a4ea015ec703c84a74ea775d88b99 (diff) | |
| download | armnn-edeeb168c6d2612d5b934158bf18c35c843dfc3e.tar.gz | |
IVGCVSW-3264 Add Unit Test for Dilated Convolution2d to armnn
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I60da1414ab41ea196cdebc27f0f014a502274fa8
Diffstat (limited to 'src/backends/backendsCommon/test/Conv2dTestImpl.hpp')
| -rw-r--r-- | src/backends/backendsCommon/test/Conv2dTestImpl.hpp | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/src/backends/backendsCommon/test/Conv2dTestImpl.hpp b/src/backends/backendsCommon/test/Conv2dTestImpl.hpp index 73b170e930..5f66d2ec85 100644 --- a/src/backends/backendsCommon/test/Conv2dTestImpl.hpp +++ b/src/backends/backendsCommon/test/Conv2dTestImpl.hpp @@ -88,7 +88,9 @@ LayerTestResult<T, 4> SimpleConvolution2dTestImpl( 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 inputHeight = boost::numeric_cast<unsigned int>(originalInput.shape()[2]); unsigned int inputWidth = boost::numeric_cast<unsigned int>(originalInput.shape()[3]); @@ -218,6 +220,8 @@ LayerTestResult<T, 4> SimpleConvolution2dTestImpl( data.m_Parameters.m_PadBottom = padBottom; data.m_Parameters.m_BiasEnabled = biasEnabled; data.m_Parameters.m_DataLayout = layout; + data.m_Parameters.m_DilationX = dilationX; + data.m_Parameters.m_DilationY = dilationY; std::unique_ptr<armnn::IWorkload> workload = workloadFactory.CreateConvolution2d(data, info); inputHandle->Allocate(); |