File: __init__.py

package info (click to toggle)
python-opt-einsum-fx 0.1.4-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 180 kB
  • sloc: python: 664; makefile: 13
file content (15 lines) | stat: -rw-r--r-- 363 bytes parent folder | download 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
 
__version__ = "0.1.4"

from ._script import jitable
from ._opt_ein import optimize_einsums, optimize_einsums_full
from ._fuse import fuse_einsums, fuse_scalars
from ._efficient_shape_prop import EfficientShapeProp

__all__ = [
    "jitable",
    "optimize_einsums",
    "optimize_einsums_full",
    "fuse_einsums",
    "fuse_scalars",
    "EfficientShapeProp",
]