• T
    [CPU] refine cpu softmax bwd (#17534) · 7ae461eb
    tensor-tang 提交于
    * refine softmax fwd
    
    test=develop
    
    * refine cpu softmax bwd
    
    test=develop
    
    * fix batch size
    
    test=develop
    
    * fix compile issue with gpu
    
    test=develop
    
    * add value clip
    7ae461eb
softmax_impl.h 8.3 KB