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, ...@@ -78,7 +78,7 @@ inline tvm::Tensor leaky_relu(const tvm::Tensor& t,
[&](const tvm::Array<tvm::Var>& i) { [&](const tvm::Array<tvm::Var>& i) {
auto value = t(i); auto value = t(i);
auto calpha = tvm::make_const(value.type(), alpha); 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, name,
tag); 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