ArmNN
 24.08
ActivationDescriptor Struct Reference

An ActivationDescriptor for the ActivationLayer. More...

#include <Descriptors.hpp>

Inheritance diagram for ActivationDescriptor:
[legend]
Collaboration diagram for ActivationDescriptor:
[legend]

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...
 

Detailed Description

An ActivationDescriptor for the ActivationLayer.

Definition at line 36 of file Descriptors.hpp.

Constructor & Destructor Documentation

◆ ActivationDescriptor() [1/2]

Definition at line 38 of file Descriptors.hpp.

40  , m_A(0)
41  , m_B(0)
42  {}

References armnn::Sigmoid.

◆ ActivationDescriptor() [2/2]

ActivationDescriptor ( armnn::ActivationFunction  activation,
float  a = 0,
float  b = 0 
)
inline

Definition at line 44 of file Descriptors.hpp.

47  : m_Function(activation)
48  , m_A(a)
49  , m_B(b)
50  {}

Member Function Documentation

◆ operator==()

bool operator== ( const ActivationDescriptor rhs) const
inline

Definition at line 52 of file Descriptors.hpp.

53  {
54  return m_Function == rhs.m_Function && m_A == rhs.m_B && m_B == rhs.m_B;
55  }

References ActivationDescriptor::m_A, ActivationDescriptor::m_B, and ActivationDescriptor::m_Function.

Member Data Documentation

◆ m_A

float m_A

◆ m_B

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().

◆ m_Function


The documentation for this struct was generated from the following file:
armnn::ActivationDescriptor::m_A
float m_A
Alpha upper bound value used by the activation functions. (BoundedReLu, Linear, TanH,...
Definition: Descriptors.hpp:61
armnn::ActivationDescriptor::m_Function
ActivationFunction m_Function
The activation function to use (Sigmoid, TanH, Linear, ReLu, BoundedReLu, SoftReLu,...
Definition: Descriptors.hpp:59
armnn::ActivationDescriptor::m_B
float m_B
Beta lower bound value used by the activation functions. (BoundedReLu, Linear, TanH).
Definition: Descriptors.hpp:63
armnn::ActivationFunction::Sigmoid
@ Sigmoid