Commit Graph

22 Commits (c42d662e2afa05d3d20d026c3f5c1ba376aa53a4)

Author SHA1 Message Date
Jiawei Wang 8f035fb637
Add TopK Op Grad CPU&GPU Kernel test=develop (#22628)
5 years ago
Aurelius84 5cb2c74127
add register op_data_type of pad/expand_as et.al (#21718)
5 years ago
zhaoyuchen2018 3ff5cc2d5e
Fix topk compile failed on windows (#21243)
5 years ago
zhaoyuchen2018 b93870e696
Improve topk performance. (#21087)
5 years ago
Tao Luo 02270b3eb1
remove unused assert.h (#19529)
6 years ago
whs 2314f2ebb3
Make topk op support variable k. (#15044)
6 years ago
Wu Yi 856f0da0fe
Fp16 training (#14992)
6 years ago
typhoonzero da87f7a698 Revert "[Feature] Fp16 training for resnet50 (#14850)"
6 years ago
Wu Yi 3d750f9c5a
[Feature] Fp16 training for resnet50 (#14850)
6 years ago
qingqing01 c7379a7320 Fix top_k op (#14034)
6 years ago
qingqing01 41e4f7ea0a
Optimize Topk when height is large. (#13710)
6 years ago
dzhwinter 4069262f0e
Revert ""cherry picked operators changes" (#12184)" (#12747)
7 years ago
dzhwinter bf3c34960f
"cherry picked operators changes" (#12184)
7 years ago
chengduoZH d36af62c1e wrap_shfl_x_sync
7 years ago
chengduo 54797abd53
Merge pull request #10347 from chengduoZH/replace___shfl_with__shfl_sync
7 years ago
dzhwinter f63ff90b03
Fix/fp64 (#10346)
7 years ago
chengduoZH 0cc635497c merge develop
7 years ago
chengduoZH b8f7fa97b6 replace __shfl with __shfl_sync
7 years ago
Abhinav Arora c241959e48 Fix CPPLint errors in operators (#9828)
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