ARM Compute Library
17.04
|
Activation Layer Information class. More...
#include <Types.h>
Public Types | |
enum | ActivationFunction { LOGISTIC, TANH, RELU, BOUNDED_RELU, SOFT_RELU, ABS, SQUARE, SQRT, LINEAR } |
Available activation functions. More... | |
Public Member Functions | |
ActivationLayerInfo (ActivationFunction f, float a=0.0f, float b=0.0f) | |
Default Constructor. More... | |
ActivationFunction | activation () const |
float | a () const |
float | b () const |
|
strong |
|
inline |
Default Constructor.
[in] | f | The activation function to use. |
[in] | a | (Optional) The alpha parameter used by some activation functions (ActivationFunction::BOUNDED_RELU, ActivationFunction::LINEAR, ActivationFunction::TANH). |
[in] | b | (Optional) The beta parameter used by some activation functions (ActivationFunction::LINEAR, ActivationFunction::TANH). |
Definition at line 458 of file Types.h.
|
inline |