• H
    [Custom Operator] Custom op support inplace mechanism (#51620) · f824bc0d
    HongyuJia 提交于
    * init unit test commit, contains register thinking
    
    * support inplace
    
    * get inplaced x.grad
    
    * Try support inplace and hook at the same time
    
    * Support inplace, need debug
    
    * Support inplace successfully
    
    * Inplace use Tensor&, consistent with Tensor*
    
    * fix MapPlainOutputs bug
    
    * fix double grad inplace error
    f824bc0d
custom_operator.cc 44.7 KB