Commit Graph

12 Commits (5b97b8556eb21698f621504c9f7031db1de58bb5)

Author SHA1 Message Date
tensor-tang 7ae461eb13
[CPU] refine cpu softmax bwd (#17534)
6 years ago
dengkaipeng 90bd038d35 fix format. test=develop
6 years ago
dengkaipeng 6c64182709 refine softmax kernel. test=develop
6 years ago
Wu Yi fd85418329
[Feature] support mix precision training for resnet (#14899)
6 years ago
Jacek Czaja 9b0eae3023 - Removing partial specialization of sotmax for inference for GPU
6 years ago
Jacek Czaja b361579f09 - Softmax for Inference is enabled when ON_INFER is set
6 years ago
Tao Luo 5b9c62faee
Revert "Softmax op optimization for inference "
6 years ago
Jacek Czaja c1fccc29c1 - Noise adding removed for Test phase of softmax
6 years ago
dzhwinter 128adf53cb
[Speed]implement cudnn sequence softmax cudnn (#8978)
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