Commit Graph

15 Commits (3ccd8964a47f679d0739dac56de7145b67c6ec9c)

Author SHA1 Message Date
tensor-tang 6e1ee7fb57 cache softmax kernel func
6 years ago
tensor-tang d59f733551 refine softmax and use with cache
6 years ago
sneaxiy a500dfa579 rewrite ddim
6 years ago
gongweibao f1fb64b17f
Add reduce sparse tensor feature. (#14757)
6 years ago
Jacek Czaja 8bfa1fa9bb - ASUM MKL integration
7 years ago
Jacek Czaja 9b0eae3023 - Removing partial specialization of sotmax for inference for GPU
7 years ago
Jacek Czaja 513bb6c151 Squashing MKL based softmax for inference
7 years ago
Jacek Czaja b361579f09 - Softmax for Inference is enabled when ON_INFER is set
7 years ago
Tao Luo 5b9c62faee
Revert "Softmax op optimization for inference "
7 years ago
Jacek Czaja d332326847 - Added unit tests for softmax is_test=True op
7 years ago
Jacek Czaja c1fccc29c1 - Noise adding removed for Test phase of softmax
7 years ago
Kexin Zhao b2a1c9e8b7 Add float16 support to non-cudnn softmax op on GPU (#9686)
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