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