This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
Repositories
Datasets
Forum
实训
竞赛
大数据
应用
Register
Sign In
OpenI
/
MegEngine
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
31
Wiki
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
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
df356635b7
commit
dbaf84b0ef
1 changed files
with
1 additions
and
0 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+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");
Write
Preview
Loading…
Cancel
Save