• Q
    Softmax grad op (#3164) · d953611e
    Qiao Longfei 提交于
    * init softmax grad op
    
    * add compute code
    
    * export Backward to python
    
    * update test ,export op.type to python
    
    * update python test, fix compute bug
    
    * update unit test
    
    * use eigen
    
    * optimize eigen code
    
    * add gpu test
    
    * register softmax_grad GPU kernel and fix test bug
    
    * typo
    
    * follow comments
    d953611e
softmax_op.h 3.2 KB