diff options
| author | Ferran Balaguer <ferran.balaguer@arm.com> | 2019-06-17 12:23:52 +0100 |
|---|---|---|
| committer | Ferran Balaguer Arm <ferran.balaguer@arm.com> | 2019-06-17 14:53:05 +0000 |
| commit | e52211e1544a30d24b29523c389116a9e4446e8c (patch) | |
| tree | 8d2cce39db4c89509a8b6c8569e85edb6d348b4b /src/backends/reference | |
| parent | 0b679f03ead8848edc225d44afbbcc54d112e433 (diff) | |
| download | armnn-e52211e1544a30d24b29523c389116a9e4446e8c.tar.gz | |
IVGCVSW-3286 Add epsilon Reference workload support
Change-Id: I5cabbf9d1ef3858be68d6820d14845e512128c5b
Signed-off-by: Ferran Balaguer <ferran.balaguer@arm.com>
Diffstat (limited to 'src/backends/reference')
| -rw-r--r-- | src/backends/reference/test/RefLayerTests.cpp | 3 | ||||
| -rw-r--r-- | src/backends/reference/workloads/RefL2NormalizationWorkload.cpp | 4 |
2 files changed, 6 insertions, 1 deletions
diff --git a/src/backends/reference/test/RefLayerTests.cpp b/src/backends/reference/test/RefLayerTests.cpp index 7540f1d4c1..cf4d9fd6f9 100644 --- a/src/backends/reference/test/RefLayerTests.cpp +++ b/src/backends/reference/test/RefLayerTests.cpp @@ -528,6 +528,9 @@ ARMNN_AUTO_TEST_CASE(L2Normalization2dUint8Nhwc, L2Normalization2dUint8Test, arm ARMNN_AUTO_TEST_CASE(L2Normalization3dUint8Nhwc, L2Normalization3dUint8Test, armnn::DataLayout::NHWC) ARMNN_AUTO_TEST_CASE(L2Normalization4dUint8Nhwc, L2Normalization4dUint8Test, armnn::DataLayout::NHWC) +ARMNN_AUTO_TEST_CASE(L2NormalizationDefaultEpsilon, L2NormalizationDefaultEpsilonTest, armnn::DataLayout::NCHW) +ARMNN_AUTO_TEST_CASE(L2NormalizationNonDefaultEpsilon, L2NormalizationNonDefaultEpsilonTest, armnn::DataLayout::NCHW) + // Pad ARMNN_AUTO_TEST_CASE(PadFloat322d, PadFloat322dTest) ARMNN_AUTO_TEST_CASE(PadFloat323d, PadFloat323dTest) diff --git a/src/backends/reference/workloads/RefL2NormalizationWorkload.cpp b/src/backends/reference/workloads/RefL2NormalizationWorkload.cpp index ce5699ef0b..3b2ab50c8b 100644 --- a/src/backends/reference/workloads/RefL2NormalizationWorkload.cpp +++ b/src/backends/reference/workloads/RefL2NormalizationWorkload.cpp @@ -61,7 +61,9 @@ RefL2NormalizationWorkload::RefL2NormalizationWorkload( unsigned int index = dataLayout.GetIndex(inputInfo.GetShape(), n, c, h, w); - const float scale = 1.0f / sqrtf(reduction); + float maximum = reduction < m_Data.m_Parameters.m_Eps ? m_Data.m_Parameters.m_Eps : reduction; + + const float scale = 1.0f / sqrtf(maximum); (*inputDecoder)[index]; (*outputEncoder)[index]; |