diff options
Diffstat (limited to 'src/graph/expression_operators.h')
-rwxr-xr-x | src/graph/expression_operators.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/graph/expression_operators.h b/src/graph/expression_operators.h index 78aed834..0205fe86 100755 --- a/src/graph/expression_operators.h +++ b/src/graph/expression_operators.h @@ -17,6 +17,9 @@ Expr sigmoid(const std::vector<Expr>&); Expr swish(Expr a); Expr swish(const std::vector<Expr>&); +Expr gelu(Expr a); +Expr gelu(const std::vector<Expr>&); + Expr tanh(const std::vector<Expr>&); template <typename... Args> |