“b0398c8e9db4f4608fd57b7b42df03558fb23366”上不存在“python/paddle/geometric/sampling/neighbors.py”
  • 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
test_softmax_op.py 2.7 KB