[Layer] Add missing activation types
authorSeoHyungjun <hyungjun.seo@samsung.com>
Wed, 3 Jul 2024 02:47:41 +0000 (11:47 +0900)
committerJijoong Moon <jijoong.moon@samsung.com>
Thu, 18 Jul 2024 04:16:31 +0000 (13:16 +0900)
Some activation types were missing from EnumList.
Added missing types to EnumList.

Changed the order of ActivationType and EnumList to be the same.

**Self evaluation:**
1. Build test:  [X]Passed [ ]Failed [ ]Skipped
2. Run test:  [X]Passed [ ]Failed [ ]Skipped

Signed-off-by: SeoHyungjun <hyungjun.seo@samsung.com>
nntrainer/layers/common_properties.h

index 8844783add5c6e506dc7295a0a1b167564ef7ef0..c5a514b6376b360e58fae9263c447acb43d31c89 100644 (file)
@@ -909,14 +909,16 @@ public:
 struct ActivationTypeInfo {
   using Enum = nntrainer::ActivationType;
   static constexpr std::initializer_list<Enum> EnumList = {
-    Enum::ACT_TANH,    Enum::ACT_SIGMOID,    Enum::ACT_RELU,
-    Enum::ACT_SOFTMAX, Enum::ACT_LEAKY_RELU, Enum::ACT_SWISH,
-    Enum::ACT_GELU,    Enum::ACT_TANH_GELU,  Enum::ACT_SIGMOID_GELU,
-    Enum::ACT_NONE,    Enum::ACT_UNKNOWN};
+    Enum::ACT_TANH,         Enum::ACT_SIGMOID, Enum::ACT_RELU,
+    Enum::ACT_SWISH,        Enum::ACT_GELU,    Enum::ACT_TANH_GELU,
+    Enum::ACT_SIGMOID_GELU, Enum::ACT_SOFTMAX, Enum::ACT_SOFTPLUS,
+    Enum::ACT_LEAKY_RELU,   Enum::ACT_ELU,     Enum::ACT_SELU,
+    Enum::ACT_MISH,         Enum::ACT_NONE,    Enum::ACT_UNKNOWN};
 
   static constexpr const char *EnumStr[] = {
-    "tanh", "sigmoid",   "relu",         "softmax", "leaky_relu", "swish",
-    "gelu", "tanh_gelu", "sigmoid_gelu", "none",    "unknown"};
+    "tanh",      "sigmoid",      "relu",    "swish",    "gelu",
+    "tanh_gelu", "sigmoid_gelu", "softmax", "softplus", "leaky_relu",
+    "elu",       "selu",         "mish",    "none",     "unknown"};
 };
 
 /**