• C
    [Phi] Reduce grad (#40263) · f452ad5c
    chentianyu03 提交于
    * add reduce_sum grad kernel
    
    * add reduce_grad
    
    * modify reduce grad
    
    * update reduce grad functions
    
    * fix build error
    
    * add argument mapping
    
    * move cast input after grad
    
    * add dims.size=1 cpu reduce_sum grad compute method
    
    * update reduce grad GPU
    
    * remove raw reduce_sum_grad kernel
    
    * modify header files
    
    * add namespace funcs for reduce_grad_funcstions
    f452ad5c
reduce_sum_grad_kernel.cu 2.9 KB