• L
    add diag_embed op (#23385) · 87d8dc3d
    Li Fuchen 提交于
    * add diag_embed op, test=develop
    
    * add TestCase of diag_embed API
    
    * Modified diag embed python API teastcase from dygraph to static graph, test=develop
    
    * delete useless log and trigger ci, test=develop
    
    * modified float16 of diag_embed, test=develop
    
    * modified en doc of diag_embed
    
    * trigger ci, test=develop
    
    * add fp16 in dtype check of python API, test=develop
    
    * modified __init__ and fix a big, test=develop
    
    * modified a test bug of test_bicubic_interp_op and test_trilinear_interp_op, test=develop
    
    * modified to use one kernel on cpu and cuda, test=develop
    87d8dc3d
diag_embed_op.h 4.1 KB