Commit c24e82e3 by Yuanbo Ye Committed by Tianqi Chen

[TOPI] fix bug of leaky_relu type issue (#1076)

Signed-off-by: Janboe Ye <yeyuanbo@xiaomi.com>
parent 9fb26056
......@@ -78,7 +78,7 @@ inline tvm::Tensor leaky_relu(const tvm::Tensor& t,
[&](const tvm::Array<tvm::Var>& i) {
auto value = t(i);
auto calpha = tvm::make_const(value.type(), alpha);
return tvm::select(value > 0, value, value * alpha);
return tvm::select(value > 0, value, value * calpha);
},
name,
tag);
......
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