• J
    【Prim】Enhance gather vjp (#50786) · dca3a099
    Jiabin Yang 提交于
    * tmp gather vjp
    
    * support gather
    
    * remove useless code
    
    * fix compiling error
    
    * fix ut
    
    * add eager test
    
    * add eager test
    
    * add seed
    
    * fix cpu error
    
    * fix transpose op compat
    
    * remove tensor index case
    
    * fix prim_cinn
    
    * fix ut
    
    * add gather composite
    dca3a099
legacy_backward.yaml 59.2 KB