typhoonzero
|
54e9fd3f61
|
fix cudnn enforce
|
7 years ago |
dzhwinter
|
2b9ef7e249
|
"fix"
|
7 years ago |
|
dzhwinter
|
75d8e8ca33
|
"fix compiled in manylinux"
|
7 years ago |
|
dzhwinter
|
4777aec9be
|
"done"
|
7 years ago |
dzhwinter
|
7971d4a310
|
Feature/deterministic (#11205)
* "fix deterministic"
* "fix ci"
* "fix init"
|
7 years ago |
Kexin Zhao
|
187ba08789
|
enable tensor core for conv cudnn
|
7 years ago |
|
Kexin Zhao
|
446d54f5c3
|
update
|
7 years ago |
|
Kexin Zhao
|
ffa22a5f90
|
fix scaling param type
|
7 years ago |
|
Kexin Zhao
|
e870947cfd
|
fix batch norm fp16 param type
|
7 years ago |
|
Kexin Zhao
|
a13ec3432a
|
fix test error
|
7 years ago |
|
Kexin Zhao
|
e4de5dc347
|
add conv2d fp16 support
|
7 years ago |
|
dzhwinter
|
128adf53cb
|
[Speed]implement cudnn sequence softmax cudnn (#8978)
* "add softmax cudnn functor support"
* "add testing"
* "refine cmakelist"
* "sequence softmax forward speed up"
* "add softmax grad"
* "fix sequence softmax test"
* "add double precision'
* "fix softmax test"
* "add softmax cudnn support"
* "fix softmax cudnn test"
* "add softmax to nn.py"
* "fix compile bug"
* "refine cmakelist"
* "fix ci"
* "fix based on comment"
* "fix based on comments"
* "fix ci"
|
7 years ago |
pzelazko-intel
|
8c71adaa8c
|
MKLDNN conv2d kernel added (#8451)
* MKLDNN conv2 OP kernel added
* TODOs added
* mkldnn conv2d OP refactor
* CanCUDNNBeUsed and CanMKLDNNBeUsed moved
|
7 years ago |
qingqing01
|
24509f4af9
|
Fix the grammar in copyright. (#8403)
|
7 years ago |
Yi Wang
|
fc374821dd
|
Correct #include path
|
7 years ago |
|
Yi Wang
|
90648f336d
|
Move file to fluid/; Edit CMakeLists.txt
|
7 years ago |
