• H
    Add center Loss Op Support (#18681) · 24f85431
    HaoRen 提交于
    * support center loss
    * change tensor copy  api to high level api tensorcopy
    
    * test=develop rewrite the center_loss cuda_kernel to make it faster
    and add document of the center loss api,also update test function
    
    * test=document_preview test=develop
    update document of center loss
    
    * test=document_preview test=develop
    modify API.spec modify test code remove nouse const_cast
    24f85431
center_loss_op.cc 5.9 KB