diff options
author | jimfly01 <jim.flynn@arm.com> | 2018-10-26 12:56:21 +0100 |
---|---|---|
committer | Jim Flynn Arm <jim.flynn@arm.com> | 2018-10-30 17:21:00 +0000 |
commit | d84216a013445e86183e39c8b5b904836c71a95b (patch) | |
tree | 5d9bc287d0a71829e3b9db6e7945f654af0ac29e /src/backends/test/LayerTests.hpp | |
parent | db6a7fa7fc9f696beacbf265a154f8e4220450dc (diff) | |
download | armnn-d84216a013445e86183e39c8b5b904836c71a95b.tar.gz |
IVGCVSW-2074: Updated the DepthwiseConvolution2dTest for NHWC
Change-Id: If2befd3bd75da70c57f924cbe18fa0a47bf7a27f
Diffstat (limited to 'src/backends/test/LayerTests.hpp')
-rw-r--r-- | src/backends/test/LayerTests.hpp | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/src/backends/test/LayerTests.hpp b/src/backends/test/LayerTests.hpp index fec73d08f4..0e45024e22 100644 --- a/src/backends/test/LayerTests.hpp +++ b/src/backends/test/LayerTests.hpp @@ -72,7 +72,9 @@ LayerTestResult<float, 4> Convolution1dTest(armnn::IWorkloadFactory& workloadF LayerTestResult<uint8_t, 4> Convolution1dUint8Test(armnn::IWorkloadFactory& workloadFactory, bool biasEnabled); -LayerTestResult<float, 4> DepthwiseConvolution2dTest(armnn::IWorkloadFactory& workloadFactory, bool biasEnabled); +LayerTestResult<float, 4> DepthwiseConvolution2dTest(armnn::IWorkloadFactory& workloadFactory, + bool biasEnabled, + const armnn::DataLayoutIndexed& layout); LayerTestResult<float, 4> DepthwiseConvolution2dDepthNhwcTest(armnn::IWorkloadFactory& workloadFactory, bool biasEnabled); @@ -327,7 +329,8 @@ LayerTestResult<uint8_t, 4> SimpleConvolution2d3x3Uint8Test(armnn::IWorkloadFact const armnn::DataLayoutIndexed& layout); LayerTestResult<uint8_t, 4> DepthwiseConvolution2dUint8Test(armnn::IWorkloadFactory& workloadFactory, - bool biasEnabled); + bool biasEnabled, + const armnn::DataLayoutIndexed& layout); LayerTestResult<uint8_t, 4> DepthwiseConvolution2dDepthMul1Uint8Test(armnn::IWorkloadFactory& workloadFactory, bool biasEnabled); |