“5d0a2de595c2754429bf2232ae659ca10367b184”上不存在“develop/api_doc/v2/run_logic.html”
  • 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.h 1.2 KB