diff options
author | Francis Murtagh <francis.murtagh@arm.com> | 2018-10-05 14:08:48 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-10 16:16:58 +0100 |
commit | 043d0d0e07e0a29648894cb9ba68993e540bb94d (patch) | |
tree | 8f757c83d11ba0472f6b636f8ce8fa5b2f3e620e /src/backends/test/LayerTests.cpp | |
parent | 7c22c70418192cf5803ae60dd2da041b0a6aafe5 (diff) | |
download | armnn-043d0d0e07e0a29648894cb9ba68993e540bb94d.tar.gz |
IVGCVSW-1860 Support NHWC for Pooling2D
Change-Id: I9059a292e47867df82a5efbba5808fd264949ec9
Diffstat (limited to 'src/backends/test/LayerTests.cpp')
-rw-r--r-- | src/backends/test/LayerTests.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/backends/test/LayerTests.cpp b/src/backends/test/LayerTests.cpp index a7fb6a824e..17f3ae12e1 100644 --- a/src/backends/test/LayerTests.cpp +++ b/src/backends/test/LayerTests.cpp @@ -4929,6 +4929,11 @@ LayerTestResult<float, 4> SimpleAveragePooling2dTest(armnn::IWorkloadFactory& wo return SimpleAveragePooling2dTestCommon<float>(workloadFactory); } +LayerTestResult<float, 4> SimpleAveragePooling2dNhwcTest(armnn::IWorkloadFactory& workloadFactory) +{ + return SimpleAveragePooling2dNhwcTestCommon<float>(workloadFactory); +} + LayerTestResult<uint8_t, 4> SimpleAveragePooling2dUint8Test(armnn::IWorkloadFactory& workloadFactory) { return SimpleAveragePooling2dTestCommon<uint8_t>(workloadFactory, 0.5, -1); |