Browse Source

feat(imperative): add cond_take opr

GitOrigin-RevId: 5272e6fa71
tags/v1.0.0-rc1
Megvii Engine Team Xinran Xu 4 years ago
parent
commit
dbaf84b0ef
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      dnn/src/naive/cond_take/opr_impl.cpp

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

@@ -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");


Loading…
Cancel
Save