Skip to content
Snippets Groups Projects
Commit c24e82e3 authored by Yuanbo Ye's avatar Yuanbo Ye Committed by Tianqi Chen
Browse files

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


Signed-off-by: default avatarJanboe Ye <yeyuanbo@xiaomi.com>
parent 9fb26056
No related branches found
No related tags found
No related merge requests found
......@@ -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);
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment