From 539b44dbd620c9f793f84933c1bcc51ce3ff085e Mon Sep 17 00:00:00 2001 From: Matteo Martincigh Date: Mon, 1 Oct 2018 09:26:39 +0100 Subject: IVGCVSW-1863 Unit tests for NHWC L2Normalization * Added NHWC unit test implementation * Programmatically selected the channel dimension when creating the ACL Normalization layer info * Set the input/output data layout in the constructor of the L2Normalization workload Change-Id: Ie69f1a360022c29d1a3a3808c1f26b69243fa8f9 --- src/backends/test/LayerTests.hpp | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'src/backends/test/LayerTests.hpp') diff --git a/src/backends/test/LayerTests.hpp b/src/backends/test/LayerTests.hpp index 3e5bb3d31d..e4ebaff527 100644 --- a/src/backends/test/LayerTests.hpp +++ b/src/backends/test/LayerTests.hpp @@ -249,6 +249,11 @@ LayerTestResult L2Normalization2dTest(armnn::IWorkloadFactory& workloa LayerTestResult L2Normalization3dTest(armnn::IWorkloadFactory& workloadFactory); LayerTestResult L2Normalization4dTest(armnn::IWorkloadFactory& workloadFactory); +LayerTestResult L2Normalization1dNhwcTest(armnn::IWorkloadFactory& workloadFactory); +LayerTestResult L2Normalization2dNhwcTest(armnn::IWorkloadFactory& workloadFactory); +LayerTestResult L2Normalization3dNhwcTest(armnn::IWorkloadFactory& workloadFactory); +LayerTestResult L2Normalization4dNhwcTest(armnn::IWorkloadFactory& workloadFactory); + LayerTestResult ConstantTest(armnn::IWorkloadFactory& workloadFactory); LayerTestResult ConstantTestUint8(armnn::IWorkloadFactory& workloadFactory); -- cgit v1.2.1