• X
    [EinsumOp] Optimize the backward speed of EinsumOp (#42663) · 71b046cd
    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
    
    * Compatible with v2.3 EinsumOp
    
    * remove dispensable
    71b046cd
test_einsum_op.py 4.7 KB