xref: /aosp_15_r20/external/tensorflow/third_party/opt_einsum.BUILD (revision b6fb3261f9314811a0f4371741dbb8839866f948) 
1# Description:
2#   opt_einsum is a library for optimizing tensor contraction order for einsum.
3
4licenses(["notice"])  # MIT
5
6exports_files(["LICENSE"])
7
8py_library(
9    name = "opt_einsum",
10    srcs = [
11        "opt_einsum/__init__.py",
12        "opt_einsum/_version.py",
13        "opt_einsum/backends/__init__.py",
14        "opt_einsum/backends/cupy.py",
15        "opt_einsum/backends/dispatch.py",
16        "opt_einsum/backends/tensorflow.py",
17        "opt_einsum/backends/theano.py",
18        "opt_einsum/backends/torch.py",
19        "opt_einsum/blas.py",
20        "opt_einsum/compat.py",
21        "opt_einsum/contract.py",
22        "opt_einsum/helpers.py",
23        "opt_einsum/parser.py",
24        "opt_einsum/paths.py",
25        "opt_einsum/sharing.py",
26    ],
27    srcs_version = "PY3",
28    visibility = ["//visibility:public"],
29)
30