“b550d68131b672741ae97422b7cc1211796c6e43”上不存在“doc/api/v2/fluid/optimizer.html”
  • C
    [Phi] Move softmax with cross entropy kernel into phi (#40832) · e6ec98fe
    Chen Weihang 提交于
    * add cross_entropy_with_softmax phi kernel
    
    * remove softmax_with_cross_entropy kernel
    
    * add softmax_with_cross_entropy grad kernel
    
    * remove original op kernel
    
    * refine cross entropy impl
    
    * fix pointer error
    
    * revert kernel cu change
    
    * fix xpu failed
    
    * fix cinn failed
    
    * fix npu failed
    
    * add forward sig
    
    * add check_nan_inf for pt kernel
    
    * remove repeat cmake item
    
    * fix unittest error
    e6ec98fe
softmax.h 1.8 KB