• J
    【Prim】Reshape, transpose, cast vjp (#50778) · ab1b6303
    Jiabin Yang 提交于
    * support transpose and reshape
    
    * support reshpe, transpose, cast vjp
    
    * merge develop
    
    * recover unused file
    
    * remove prim base
    
    * support problem
    
    * remove additional status settting
    
    * remove additional status settting
    
    * fix ut
    
    * fix ut
    
    * fix ut
    
    * fix no grad branch
    
    * add more test
    
    * disable fp16 in cpu
    
    * fix test
    ab1b6303
prim_manual_api.h 1.4 KB