From 55a97bc2605fc1246a9a1f7ee89cde415496a1ba Mon Sep 17 00:00:00 2001 From: narpra01 Date: Tue, 2 Oct 2018 14:35:53 +0100 Subject: IVGCVSW-1920 Unittests for NHWC Normalization Workloads and Layer Change-Id: Iea941c1747454f5a4342351e4e82b10ffb9ccbbd --- src/backends/test/LayerTests.cpp | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'src/backends/test/LayerTests.cpp') diff --git a/src/backends/test/LayerTests.cpp b/src/backends/test/LayerTests.cpp index 55f4a5c5f9..267a8d6bad 100644 --- a/src/backends/test/LayerTests.cpp +++ b/src/backends/test/LayerTests.cpp @@ -520,6 +520,13 @@ LayerTestResult SimpleNormalizationWithinTest(armnn::IWorkloadFactory& return SimpleNormalizationTestImpl(workloadFactory, normChannel, normMethod); } +LayerTestResult SimpleNormalizationAcrossNhwcTest(armnn::IWorkloadFactory& workloadFactory) +{ + auto normMethod = armnn::NormalizationAlgorithmMethod::LocalBrightness; + auto normChannel = armnn::NormalizationAlgorithmChannel::Across; + return SimpleNormalizationNhwcClNeonTestImpl(workloadFactory, normChannel, normMethod); +} + LayerTestResult SimpleSoftmaxTest(armnn::IWorkloadFactory& workloadFactory, float beta) { return SimpleSoftmaxTestImpl(workloadFactory, beta); -- cgit v1.2.1