24.08
|
An ActivationDescriptor for the ActivationLayer. More...
#include <Descriptors.hpp>
Public Member Functions | |
ActivationDescriptor () | |
ActivationDescriptor (armnn::ActivationFunction activation, float a=0, float b=0) | |
bool | operator== (const ActivationDescriptor &rhs) const |
Public Member Functions inherited from BaseDescriptor | |
virtual bool | IsNull () const |
virtual | ~BaseDescriptor ()=default |
Public Attributes | |
ActivationFunction | m_Function |
The activation function to use (Sigmoid, TanH, Linear, ReLu, BoundedReLu, SoftReLu, LeakyReLu, Abs, Sqrt, Square, Elu). More... | |
float | m_A |
Alpha upper bound value used by the activation functions. (BoundedReLu, Linear, TanH, Elu). More... | |
float | m_B |
Beta lower bound value used by the activation functions. (BoundedReLu, Linear, TanH). More... | |
An ActivationDescriptor for the ActivationLayer.
Definition at line 36 of file Descriptors.hpp.
|
inline |
Definition at line 38 of file Descriptors.hpp.
References armnn::Sigmoid.
|
inline |
Definition at line 44 of file Descriptors.hpp.
|
inline |
Definition at line 52 of file Descriptors.hpp.
References ActivationDescriptor::m_A, ActivationDescriptor::m_B, and ActivationDescriptor::m_Function.
float m_A |
Alpha upper bound value used by the activation functions. (BoundedReLu, Linear, TanH, Elu).
Definition at line 61 of file Descriptors.hpp.
Referenced by armnn::ConvertActivationDescriptorToAclActivationLayerInfo(), ConvertReluToTosaOperator(), ActivationDescriptor::operator==(), armnn_driver::ProcessActivation(), and StringifyLayerParameters< ActivationDescriptor >::Serialize().
float m_B |
Beta lower bound value used by the activation functions. (BoundedReLu, Linear, TanH).
Definition at line 63 of file Descriptors.hpp.
Referenced by armnn::ConvertActivationDescriptorToAclActivationLayerInfo(), ActivationDescriptor::operator==(), armnn_driver::ProcessActivation(), and StringifyLayerParameters< ActivationDescriptor >::Serialize().
ActivationFunction m_Function |
The activation function to use (Sigmoid, TanH, Linear, ReLu, BoundedReLu, SoftReLu, LeakyReLu, Abs, Sqrt, Square, Elu).
Definition at line 59 of file Descriptors.hpp.
Referenced by armnn::BuildAddMulAddTensorInfoLists(), ClSqrtWorkload::ClSqrtWorkload(), armnn::ConvertActivationDescriptorToAclActivationLayerInfo(), ConvertGeluToTosaOperator(), ConvertReluToTosaOperator(), RefLayerSupport::IsActivationSupported(), NeonSqrtWorkload::NeonSqrtWorkload(), ActivationDescriptor::operator==(), armnn_driver::ProcessActivation(), and StringifyLayerParameters< ActivationDescriptor >::Serialize().