diff options
Diffstat (limited to 'include/armnn/Descriptors.hpp')
| -rw-r--r-- | include/armnn/Descriptors.hpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp index 9479db3571..dd1991d569 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -424,9 +424,12 @@ struct NormalizationDescriptor struct L2NormalizationDescriptor { L2NormalizationDescriptor() - : m_DataLayout(DataLayout::NCHW) + : m_Eps(1e-12f) + , m_DataLayout(DataLayout::NCHW) {} + /// Used to avoid dividing by zero. + float m_Eps; /// The data layout to be used (NCHW, NHWC). DataLayout m_DataLayout; }; |