diff options
| author | Francis Murtagh <francis.murtagh@arm.com> | 2018-10-04 16:03:07 +0100 |
|---|---|---|
| committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-10 16:16:58 +0100 |
| commit | d59116ecb54c5bfe828d82ea0bc3367bc9b8c5dd (patch) | |
| tree | 782d777544416eca8dcf8b924b710a395a75ad36 /src/backends/test/LayerTests.hpp | |
| parent | de9011bc446d767932b6fec356f65791dff685e5 (diff) | |
| download | armnn-d59116ecb54c5bfe828d82ea0bc3367bc9b8c5dd.tar.gz | |
IVGCVSW-1889 - Unit test Convolution2d with NHWC
* Added simple convolution Unit test
* Set the data layout correctly in workloads
Change-Id: Ie71b8415f6abc392a84900fc4438b7416fbb558a
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 e4ebaff527..12bcdd8bbc 100644 --- a/src/backends/test/LayerTests.hpp +++ b/src/backends/test/LayerTests.hpp @@ -50,10 +50,13 @@ struct LayerTestResult }; LayerTestResult<float, 4> SimpleConvolution2d3x5Test(armnn::IWorkloadFactory& workloadFactory, - bool biasEnabled); + bool biasEnabled); LayerTestResult<float, 4> SimpleConvolution2d3x3Test(armnn::IWorkloadFactory& workloadFactory, - bool biasEnabled); + bool biasEnabled); + +LayerTestResult<float, 4> SimpleConvolution2d3x3NhwcTest(armnn::IWorkloadFactory& workloadFactory, + bool biasEnabled); LayerTestResult<float, 4> Convolution2dAsymmetricPaddingLargerThanHalfKernelSizeTest(armnn::IWorkloadFactory& workloadFactory); |