• D
    [Speed]implement cudnn sequence softmax cudnn (#8978) · 128adf53
    dzhwinter 提交于
    * "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"
    128adf53
softmax_op.cc 5.9 KB