1. 25 5月, 2022 1 次提交
    • 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
  2. 10 5月, 2022 1 次提交
  3. 29 4月, 2022 1 次提交
    • X
      Make einsum_v2 support multi-operands (#42327) · 32cae24c
      xiongkun 提交于
      * Extend python einsum interface to make einsum_v2 support multi-operands and switch it to default.
      
      * add opt_einsum dependence
      
      * add yaml and support eager model
      
      * fix by code review
      32cae24c
  4. 26 4月, 2022 1 次提交