• S
    [phi] Update graph_send_recv OP (#40509) · 67b46e45
    Siming Dai 提交于
    * add out_size shape for graph_send_recv
    
    * fix bug in register kernel: no const int& support
    
    * add out_size in infermeta
    
    * change unittest
    
    * fix unittest
    
    * fix out_size default value
    
    * fix doc
    
    * delete arg mapping
    
    * add sig
    
    * move -1 to 0
    
    * move -1 to 0
    67b46e45
graph_send_recv_grad_kernel.h 1.3 KB