From 077cddbe9e956c6740557a9add499385f235c384 Mon Sep 17 00:00:00 2001 From: Teresa Charlin Date: Fri, 15 Sep 2023 15:19:21 +0100 Subject: IVGCVSW-8055 Add support for GELU activation function. * Add support to CpuRef, CpuAcc and GpuAcc * Add support to tflite parser, classic and opaque tflite delegates * Add support to serializer and deserializer * Add Unit tests Signed-off-by: Teresa Charlin Change-Id: Ibc60ef2ef2a051e6d9af6e15d24c46316ec19de4 --- include/armnn/Types.hpp | 3 ++- include/armnn/TypesUtils.hpp | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) (limited to 'include') diff --git a/include/armnn/Types.hpp b/include/armnn/Types.hpp index 933e7da412..d87e7f7147 100644 --- a/include/armnn/Types.hpp +++ b/include/armnn/Types.hpp @@ -96,7 +96,8 @@ enum class ActivationFunction Sqrt = 8, Square = 9, Elu = 10, - HardSwish = 11 + HardSwish = 11, + Gelu = 12 }; enum class ArgMinMaxFunction diff --git a/include/armnn/TypesUtils.hpp b/include/armnn/TypesUtils.hpp index ca098f60fb..b5f224ed75 100644 --- a/include/armnn/TypesUtils.hpp +++ b/include/armnn/TypesUtils.hpp @@ -44,6 +44,7 @@ constexpr char const* GetActivationFunctionAsCString(ActivationFunction activati case ActivationFunction::Square: return "Square"; case ActivationFunction::Elu: return "Elu"; case ActivationFunction::HardSwish: return "HardSwish"; + case ActivationFunction::Gelu: return "Gelu"; default: return "Unknown"; } } -- cgit v1.2.1