diff options
Diffstat (limited to 'src/core/Utils.cpp')
| -rw-r--r-- | src/core/Utils.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/core/Utils.cpp b/src/core/Utils.cpp index 904362e1b4..48eb8b9bf5 100644 --- a/src/core/Utils.cpp +++ b/src/core/Utils.cpp @@ -177,7 +177,8 @@ const std::string &string_from_activation_func(ActivationLayerInfo::ActivationFu { ActivationLayerInfo::ActivationFunction::SQUARE, "SQUARE" }, { ActivationLayerInfo::ActivationFunction::TANH, "TANH" }, { ActivationLayerInfo::ActivationFunction::IDENTITY, "IDENTITY" }, - { ActivationLayerInfo::ActivationFunction::HARD_SWISH, "HARD_SWISH" } + { ActivationLayerInfo::ActivationFunction::HARD_SWISH, "HARD_SWISH" }, + { ActivationLayerInfo::ActivationFunction::GELU, "GELU" } }; |