• G
    Fix gradients with ignore_idx in softmax_with_cross_entropy (#28622) · 110febdc
    Guo Sheng 提交于
    * Fix gradients with ignore_idx in softmax_with_cross_entropy.
    test=develop
    
    * Fix gradients with ignore_idx in softmax_with_cross_entropy on cpu.
    Remove softmax_with_cross_entropy from op_threshold_white_list.
    test=develop
    
    * Fix test_softmax_cross_entropy_op.py.
    test=develop
    110febdc
softmax_with_cross_entropy_op.cu 21.5 KB