1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
|
"""Compute backends for opt_einsum."""
# Backends
from opt_einsum.backends.cupy import to_cupy
from opt_einsum.backends.dispatch import (
build_expression,
evaluate_constants,
get_func,
has_backend,
has_einsum,
has_tensordot,
)
from opt_einsum.backends.tensorflow import to_tensorflow
from opt_einsum.backends.theano import to_theano
from opt_einsum.backends.torch import to_torch
__all__ = [
"get_func",
"has_einsum",
"has_tensordot",
"build_expression",
"evaluate_constants",
"has_backend",
"to_tensorflow",
"to_theano",
"to_cupy",
"to_torch",
]
|
