diff options
Diffstat (limited to 'src/backends/test/LayerTests.cpp')
-rw-r--r-- | src/backends/test/LayerTests.cpp | 7 |
1 files changed, 7 insertions, 0 deletions
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<float,4> SimpleNormalizationWithinTest(armnn::IWorkloadFactory& return SimpleNormalizationTestImpl(workloadFactory, normChannel, normMethod); } +LayerTestResult<float,4> SimpleNormalizationAcrossNhwcTest(armnn::IWorkloadFactory& workloadFactory) +{ + auto normMethod = armnn::NormalizationAlgorithmMethod::LocalBrightness; + auto normChannel = armnn::NormalizationAlgorithmChannel::Across; + return SimpleNormalizationNhwcClNeonTestImpl(workloadFactory, normChannel, normMethod); +} + LayerTestResult<float,2> SimpleSoftmaxTest(armnn::IWorkloadFactory& workloadFactory, float beta) { return SimpleSoftmaxTestImpl<float>(workloadFactory, beta); |