feat(imperative): add cond_take opr

GitOrigin-RevId: 5272e6fa7181c929b2ce6cace25244154e1949a6

feat(imperative): add cond_take opr
GitOrigin-RevId: 5272e6fa7181c929b2ce6cace25244154e1949a6
dbaf84b0 · Megvii Engine Team · Xinran Xu · df356635 · dbaf84b0
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

dnn/src/naive/cond_take/opr_impl.cpp dnn/src/naive/cond_take/opr_impl.cpp +1 -0

未找到文件。
--- a/dnn/src/naive/cond_take/opr_impl.cpp
+++ b/dnn/src/naive/cond_take/opr_impl.cpp
@@ -66,6 +66,7 @@ CondTakeImpl::Output CondTakeImpl::exec(
            break; \
        }
        MEGDNN_FOREACH_COMPUTING_DTYPE(cb)
+        cb(::megdnn::dtype::Bool)
 #undef cb
        default:
            megdnn_throw("bad mask dtype");