• X
    [ Make FLAGS_einsum_opt as default ] Einsum memory optimization (#43397) · 83abec60
    xiongkun 提交于
    * change logic for optimize
    
    * modifty
    
    * optimize the backward speed of EinsumOp
    
    * add cache optimizer for einsum op
    
    * EinsumOp: fix new dygraph mode error
    
    * fix bug
    
    * change Cache->InnerCache
    
    * fix code
    
    * fix
    
    * add nan inf utils for einsum op
    
    * add as_extra
    
    * memory optimizer for einsum
    
    * update code
    83abec60
einsum.py 35.4 KB