diff options
author | Marcin Junczys-Dowmunt <junczys@amu.edu.pl> | 2018-04-11 06:46:29 +0300 |
---|---|---|
committer | Marcin Junczys-Dowmunt <junczys@amu.edu.pl> | 2018-04-11 06:46:29 +0300 |
commit | afc3bde59710c2d5b8f4b0b4b731752dc2cf9a41 (patch) | |
tree | 4b32ed16d45c98516b190ab2c2e0d51ea77a99dd /src/graph/node_operators_unary.h | |
parent | b0622430caab902efd9681ac9582d0f274168764 (diff) |
add int16 operators, attempt at memoization
Diffstat (limited to 'src/graph/node_operators_unary.h')
-rw-r--r-- | src/graph/node_operators_unary.h | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/graph/node_operators_unary.h b/src/graph/node_operators_unary.h index 8ab249a8..dc4015b2 100644 --- a/src/graph/node_operators_unary.h +++ b/src/graph/node_operators_unary.h @@ -12,9 +12,11 @@ namespace marian { struct UnaryNodeOp : public NaryNodeOp { - UnaryNodeOp(Expr a, Shape shape) : NaryNodeOp({a}, shape) {} + UnaryNodeOp(Expr a, Shape shape, Type value_type = Type::float32) + : NaryNodeOp({a}, shape, value_type) {} - UnaryNodeOp(Expr a) : NaryNodeOp({a}, a->shape()) {} + UnaryNodeOp(Expr a, Type value_type = Type::float32) + : NaryNodeOp({a}, a->shape(), value_type) {} const std::string color() { return "yellow"; } }; |