diff options
Diffstat (limited to 'include/armnn')
-rw-r--r-- | include/armnn/Types.hpp | 3 | ||||
-rw-r--r-- | include/armnn/TypesUtils.hpp | 1 |
2 files changed, 3 insertions, 1 deletions
diff --git a/include/armnn/Types.hpp b/include/armnn/Types.hpp index af331a227f..b00a8bf986 100644 --- a/include/armnn/Types.hpp +++ b/include/armnn/Types.hpp @@ -62,7 +62,8 @@ enum class ActivationFunction LeakyReLu = 6, Abs = 7, Sqrt = 8, - Square = 9 + Square = 9, + Elu = 10 }; enum class ArgMinMaxFunction diff --git a/include/armnn/TypesUtils.hpp b/include/armnn/TypesUtils.hpp index bf54c15ef8..eca8a5e233 100644 --- a/include/armnn/TypesUtils.hpp +++ b/include/armnn/TypesUtils.hpp @@ -38,6 +38,7 @@ constexpr char const* GetActivationFunctionAsCString(ActivationFunction activati case ActivationFunction::Abs: return "Abs"; case ActivationFunction::Sqrt: return "Sqrt"; case ActivationFunction::Square: return "Square"; + case ActivationFunction::Elu: return "Elu"; default: return "Unknown"; } } |