提交 101b58d1 编写于 作者: M Megvii Engine Team 提交者: Xinran Xu

fix(dnn): enable bool input to cond_take

GitOrigin-RevId: 532fa7c073849f47c22ee1f7cf4a24913d14a829
上级 eb0cc59d
......@@ -91,6 +91,7 @@ CondTakeImpl::Output CondTakeImpl::exec(
break; \
}
MEGDNN_FOREACH_COMPUTING_DTYPE(cb)
cb(::megdnn::dtype::Bool)
#undef cb
default:
megdnn_throw("bad data dtype");
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册