ArmNN
 25.11
Loading...
Searching...
No Matches
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).
float m_A
 Alpha upper bound value used by the activation functions. (BoundedReLu, Linear, TanH, Elu).
float m_B
 Beta lower bound value used by the activation functions. (BoundedReLu, Linear, TanH).

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.

39 : m_Function(ActivationFunction::Sigmoid)
40 , m_A(0)
41 , m_B(0)
42 {}

References m_A, m_B, m_Function, and armnn::Sigmoid.

Referenced by operator==().

◆ 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 {}

References m_A, m_B, and m_Function.

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, m_B, and m_Function.

Member Data Documentation

◆ m_A

◆ m_B

float m_B

◆ m_Function


The documentation for this struct was generated from the following file: