diff options
author | narpra01 <narumol.prangnawarat@arm.com> | 2018-10-02 14:35:53 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-10 16:16:58 +0100 |
commit | 55a97bc2605fc1246a9a1f7ee89cde415496a1ba (patch) | |
tree | 28043aa8cbe684f978d46c690b100000e9517312 /src/backends/test/LayerTests.cpp | |
parent | ee9e7665a5922f7ec0c5ec24d6ab2ecd88fbcfd6 (diff) | |
download | armnn-55a97bc2605fc1246a9a1f7ee89cde415496a1ba.tar.gz |
IVGCVSW-1920 Unittests for NHWC Normalization Workloads and Layer
Change-Id: Iea941c1747454f5a4342351e4e82b10ffb9ccbbd
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); |