diff options
author | Jan Eilers <jan.eilers@arm.com> | 2020-03-11 12:52:46 +0000 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2020-03-18 17:50:48 +0000 |
commit | bca73e1c82438f160364a113793d0a2195c665ac (patch) | |
tree | 0ebb17f7ce546af5bf1641e7a946d424f0f87b0a /include/armnn | |
parent | 3184c907b2420e6c66485529f336251b2b62aecf (diff) | |
download | armnn-bca73e1c82438f160364a113793d0a2195c665ac.tar.gz |
IVGCVSW-4444 Adding Elu end to end test
* implemented activation layer end to end test
* adds support for different tolerances in layer tests
* added tests for Elu (Ref, Cl, Neon)
Signed-off-by: Jan Eilers <jan.eilers@arm.com>
Change-Id: I81e28cfb4456e815bae2fb31f5c345134ff2432f
Diffstat (limited to 'include/armnn')
-rw-r--r-- | include/armnn/Descriptors.hpp | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp index f1b29cc6c7..57917261d4 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -25,15 +25,23 @@ struct ActivationDescriptor , m_B(0) {} + ActivationDescriptor(armnn::ActivationFunction activation, + float a = 0, + float b = 0) + : m_Function(activation) + , m_A(a) + , m_B(b) + {} + bool operator ==(const ActivationDescriptor &rhs) const { return m_Function == rhs.m_Function && m_A == rhs.m_B && m_B == rhs.m_B; } /// @brief The activation function to use - /// (Sigmoid, TanH, Linear, ReLu, BoundedReLu, SoftReLu, LeakyReLu, Abs, Sqrt, Square). + /// (Sigmoid, TanH, Linear, ReLu, BoundedReLu, SoftReLu, LeakyReLu, Abs, Sqrt, Square, Elu). ActivationFunction m_Function; - /// Alpha upper bound value used by the activation functions. (BoundedReLu, Linear, TanH). + /// Alpha upper bound value used by the activation functions. (BoundedReLu, Linear, TanH, Elu). float m_A; /// Beta lower bound value used by the activation functions. (BoundedReLu, Linear, TanH). float m_B; |