• K
    Softmax_cross_entropy op add axis (#16806) · a71d8fdb
    Kaipeng Deng 提交于
    * add attr axis infershape. test=develop
    
    * add CUDA kernel. test=develop
    
    * fix unittest. test=develop
    
    * fix unittest for soft_label. test=develop
    
    * fix fp16 unittest. test=develop
    
    * remove comment code. test=develop
    
    * refine test for axis. test=develop
    
    * add python api. test=develop
    
    * fix doc. test=develop
    
    * fix fp16 unittest. test=develop
    
    * fix ngraph test. test=develop
    
    * fix ENFORCE for test_imperative_transformer. test=develop
    
    * fit for ngraph test. test=develop
    
    * fix after rebase develop. test=develop
    
    * fix doc. test=develop
    
    * fix API.spec. test=develop
    
    * fix test_layers. test=develop
    
    * fix format. test=develop
    a71d8fdb
softmax_with_cross_entropy_op.cu 21.1 KB