add to handle lower or capitable case of activation function in RNN operators.

cc1a909f · Shucai Xiao · e114689e · cc1a909f
Commit cc1a909f authored Jun 14, 2019 by Shucai Xiao
Show whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

src/onnx/onnx.cpp src/onnx/onnx.cpp +6 -0

No files found.
--- a/src/onnx/onnx.cpp
+++ b/src/onnx/onnx.cpp
@@ -100,10 +100,16 @@ struct onnx_parser
    void init_actv_func()
    {
+        // Support name format of all lower case or the first letter capital
+        map_actv_funcs.insert(std::make_pair("tanh", op::tanh{}));
        map_actv_funcs.insert(std::make_pair("Tanh", op::tanh{}));
+        map_actv_funcs.insert(std::make_pair("relu", op::relu{}));
        map_actv_funcs.insert(std::make_pair("Relu", op::relu{}));
+        map_actv_funcs.insert(std::make_pair("sigmoid", op::sigmoid{}));
        map_actv_funcs.insert(std::make_pair("Sigmoid", op::sigmoid{}));
+        map_actv_funcs.insert(std::make_pair("leakyrelu", op::leaky_relu{}));
        map_actv_funcs.insert(std::make_pair("LeakyRelu", op::leaky_relu{}));
+        map_actv_funcs.insert(std::make_pair("elu", op::elu{}));
        map_actv_funcs.insert(std::make_pair("Elu", op::elu{}));
    }