Commit e114689e authored by Shucai Xiao's avatar Shucai Xiao
Browse files

update activation function name of rnn operators to be capital for the first letter

parent 84ecee26
...@@ -100,11 +100,11 @@ struct onnx_parser ...@@ -100,11 +100,11 @@ struct onnx_parser
void init_actv_func() void init_actv_func()
{ {
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{}));
} }
template <class F> template <class F>
...@@ -863,7 +863,7 @@ struct onnx_parser ...@@ -863,7 +863,7 @@ struct onnx_parser
dirct = op::rnn_direction::reverse; dirct = op::rnn_direction::reverse;
} }
std::vector<std::string> vec_names{"tanh"}; std::vector<std::string> vec_names{"Tanh"};
if(contains(attributes, "activations")) if(contains(attributes, "activations"))
{ {
auto names = attributes.at("activations").strings(); auto names = attributes.at("activations").strings();
...@@ -954,7 +954,7 @@ struct onnx_parser ...@@ -954,7 +954,7 @@ struct onnx_parser
dirct = op::rnn_direction::reverse; dirct = op::rnn_direction::reverse;
} }
std::vector<std::string> vec_names = {"sigmoid", "tanh"}; std::vector<std::string> vec_names = {"Sigmoid", "Tanh"};
if(contains(attributes, "activations")) if(contains(attributes, "activations"))
{ {
auto names = attributes.at("activations").strings(); auto names = attributes.at("activations").strings();
...@@ -1081,7 +1081,7 @@ struct onnx_parser ...@@ -1081,7 +1081,7 @@ struct onnx_parser
MIGRAPHX_THROW("LSTM: incorrect direction attribute"); MIGRAPHX_THROW("LSTM: incorrect direction attribute");
} }
std::vector<std::string> vec_names = {"sigmoid", "tanh", "tanh"}; std::vector<std::string> vec_names = {"Sigmoid", "Tanh", "Tanh"};
if(contains(attributes, "activations")) if(contains(attributes, "activations"))
{ {
auto names = attributes.at("activations").strings(); auto names = attributes.at("activations").strings();
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment