Commit Graph

20 Commits (ac9afa024b37ad00bc9d80df8a3f50fdf00a7911)

Author SHA1 Message Date
dengkaipeng 90bd038d35 fix format. test=develop
6 years ago
dengkaipeng 93701dba50 add jit kernel for softmax axis. test=develop
6 years ago
dengkaipeng 6c64182709 refine softmax kernel. test=develop
6 years ago
dengkaipeng 2ddd23dac8 fix format. test=develop
6 years ago
dengkaipeng 6cb66721d2 add cudnn support. test=develop
6 years ago
dengkaipeng 518325f1e7 add softmax_axis CPU kernel. test=develop
6 years ago
Jacek Czaja 8bfa1fa9bb - ASUM MKL integration
6 years ago
Jacek Czaja be80bb4f28 - Fix to GPU
6 years ago
Jacek Czaja 513bb6c151 Squashing MKL based softmax for inference
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 d332326847 - Added unit tests for softmax is_test=True op
6 years ago
Jacek Czaja c1fccc29c1 - Noise adding removed for Test phase of softmax
6 years ago
fengjiayi 5e7aa8c7e5 code clean
7 years ago
fengjiayi 855c9e3311 clean softmax_op code
7 years ago
fengjiayi f7bd0b227b Add unittests for softmax_op
7 years ago
fengjiayi b314a69523 make softmax supporting tensors
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