Browse Source

fix(dnn): enable bool input to cond_take

GitOrigin-RevId: 532fa7c073
tags/v1.0.0-rc1
Megvii Engine Team Xinran Xu 4 years ago
parent
commit
101b58d1ca
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

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


Loading…
Cancel
Save