Browse Source

feat(imperative): add advance indexing with bool

also fix bug to deal with empty tensor in np2tensor

GitOrigin-RevId: 2bd8152c90
tags/v1.0.0-rc1
Megvii Engine Team Xinran Xu 4 years ago
parent
commit
8aa34e4a5d
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      dnn/src/cuda/cond_take/opr_impl.cpp

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

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


Loading…
Cancel
Save