"...lm-evaluation-harness.git" did not exist on "a5e93901646eb63e8d4e388d7daaeca03e49d7b5"
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