diff options
Diffstat (limited to 'arm_compute')
-rw-r--r-- | arm_compute/core/Types.h | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/arm_compute/core/Types.h b/arm_compute/core/Types.h index 99ec44d5e0..c6c2728407 100644 --- a/arm_compute/core/Types.h +++ b/arm_compute/core/Types.h @@ -1479,7 +1479,8 @@ public: ABS, /**< Absolute ( \f$ f(x)= |x| \f$ ) */ SQUARE, /**< Square ( \f$ f(x)= x^2 \f$ )*/ SQRT, /**< Square root ( \f$ f(x) = \sqrt{x} \f$ )*/ - LINEAR /**< Linear ( \f$ f(x)= ax + b \f$ ) */ + LINEAR, /**< Linear ( \f$ f(x)= ax + b \f$ ) */ + IDENTITY /**< Identity ( \f$ f(x)= x \f$ ) */ }; ActivationLayerInfo() = default; @@ -1516,7 +1517,7 @@ public: } private: - ActivationFunction _act = { ActivationLayerInfo::ActivationFunction::LOGISTIC }; + ActivationFunction _act = { ActivationLayerInfo::ActivationFunction::IDENTITY }; float _a = {}; float _b = {}; bool _enabled = { false }; |