From afc3bde59710c2d5b8f4b0b4b731752dc2cf9a41 Mon Sep 17 00:00:00 2001 From: Marcin Junczys-Dowmunt Date: Tue, 10 Apr 2018 20:46:29 -0700 Subject: add int16 operators, attempt at memoization --- src/graph/node_operators_unary.h | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) (limited to 'src/graph/node_operators_unary.h') 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"; } }; -- cgit v1.2.3