• X
    【Prim】Add multiply,expand,div vjp rules (#49831) · 39c6765a
    Xiaoxu Chen 提交于
    * support elementwise base func
    
    * fix compiling error and add test
    
    * support vjp for div using comp
    
    * remove additional change
    
    * fix dy2st error with magic num
    
    * fix dy magic num
    
    * another magic
    
    * another magic
    
    * another magic
    
    * add skip rename strategy
    
    * support add vjp
    
    * support add with new axis cal
    
    * support sub vjp
    
    * [prim] add multiply vjp rules
    
    * [prim] add multiply vjp rules
    
    * [prim] fix no infershape with composite in _append_backward_ops
    
    * [prim] add expand vjp rule
    
    * [prim] add exp vjp rule
    
    * uncomment infer shape for reshape/sum static prim api
    
    * [prim] fix tanh nullptr error
    
    * remove some print message
    
    * fix magic number in run_program relative tests @JiaBinYang
    
    * [prim] add expand,multiply,exp vjp rules
    
    * fix only support single direction reduce error
    
    * infer reduce dims using out dims
    Co-authored-by: NJiabinYang <360788950@qq.com>
    39c6765a
utils.h 2.8 KB