提交 8aa34e4a 编写于 作者: M Megvii Engine Team 提交者: Xinran Xu

feat(imperative): add advance indexing with bool

also fix bug to deal with empty tensor in np2tensor

GitOrigin-RevId: 2bd8152c9064be7381191a251bf133f7c9f292ac
上级 101b58d1
......@@ -82,6 +82,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.
先完成此消息的编辑!
想要评论请 注册