add two more activation functions for rnn operator.

69102b29 · Shucai Xiao · cb46bbce · 69102b29
Commit 69102b29 authored Jan 23, 2019 by Shucai Xiao
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

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

No files found.
--- a/src/onnx/onnx.cpp
+++ b/src/onnx/onnx.cpp
@@ -96,6 +96,8 @@ struct onnx_parser
        actv_funcs.insert(std::make_pair("tanh", op::tanh{}));
        actv_funcs.insert(std::make_pair("relu", op::relu{}));
        actv_funcs.insert(std::make_pair("sigmoid", op::sigmoid{}));
+        actv_funcs.insert(std::make_pair("leakyrelu", op::leaky_relu{}));
+        actv_funcs.insert(std::make_pair("elu", op::elu{}));
    }
    template <class F>