From 0c64f461e071f281dab14d6b9dde85a0235d0c25 Mon Sep 17 00:00:00 2001 From: Ferran Balaguer Date: Tue, 18 Jun 2019 16:32:30 +0100 Subject: IVGCVSW-3288 Add L2Normalization epsilon Neon workload support Signed-off-by: Ferran Balaguer Change-Id: I938d97e6a2bfab8ff5f4548efeeaa43b7f5ccda2 --- src/backends/neon/test/NeonLayerTests.cpp | 3 +++ src/backends/neon/workloads/NeonL2NormalizationFloatWorkload.cpp | 4 ++-- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/src/backends/neon/test/NeonLayerTests.cpp b/src/backends/neon/test/NeonLayerTests.cpp index 7802ba2d52..425fe3a19e 100644 --- a/src/backends/neon/test/NeonLayerTests.cpp +++ b/src/backends/neon/test/NeonLayerTests.cpp @@ -404,6 +404,9 @@ ARMNN_AUTO_TEST_CASE(L2Normalization2dNhwc, L2Normalization2dTest, armnn::DataLa ARMNN_AUTO_TEST_CASE(L2Normalization3dNhwc, L2Normalization3dTest, armnn::DataLayout::NHWC) ARMNN_AUTO_TEST_CASE(L2Normalization4dNhwc, L2Normalization4dTest, armnn::DataLayout::NHWC) +ARMNN_AUTO_TEST_CASE(L2NormalizationDefaultEpsilon, L2NormalizationDefaultEpsilonTest, armnn::DataLayout::NCHW) +ARMNN_AUTO_TEST_CASE(L2NormalizationNonDefaultEpsilon, L2NormalizationNonDefaultEpsilonTest, armnn::DataLayout::NCHW) + // Floor ARMNN_AUTO_TEST_CASE(SimpleFloor, SimpleFloorTest) diff --git a/src/backends/neon/workloads/NeonL2NormalizationFloatWorkload.cpp b/src/backends/neon/workloads/NeonL2NormalizationFloatWorkload.cpp index e6249b3075..9de6c82702 100644 --- a/src/backends/neon/workloads/NeonL2NormalizationFloatWorkload.cpp +++ b/src/backends/neon/workloads/NeonL2NormalizationFloatWorkload.cpp @@ -24,7 +24,7 @@ arm_compute::Status NeonL2NormalizationWorkloadValidate(const TensorInfo& input, int axis = (descriptor.m_DataLayout == DataLayout::NCHW) ? 2 : 0; - return arm_compute::NEL2NormalizeLayer::validate(&aclInput, &aclOutput, axis); + return arm_compute::NEL2NormalizeLayer::validate(&aclInput, &aclOutput, axis, descriptor.m_Eps); } NeonL2NormalizationFloatWorkload::NeonL2NormalizationFloatWorkload(const L2NormalizationQueueDescriptor& descriptor, @@ -43,7 +43,7 @@ NeonL2NormalizationFloatWorkload::NeonL2NormalizationFloatWorkload(const L2Norma int axis = (m_Data.m_Parameters.m_DataLayout == DataLayout::NCHW) ? 2 : 0; auto layer = std::make_unique(memoryManager); - layer->configure(&input, &output, axis); + layer->configure(&input, &output, axis, m_Data.m_Parameters.m_Eps); m_Layer.reset(layer.release()); } -- cgit v1.2.1