You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
30 lines
824 B
30 lines
824 B
# Description:
|
|
# opt_einsum is a library for optimizing tensor contraction order for einsum.
|
|
|
|
licenses(["notice"]) # MIT
|
|
|
|
exports_files(["LICENSE"])
|
|
|
|
py_library(
|
|
name = "opt_einsum",
|
|
srcs = [
|
|
"opt_einsum/__init__.py",
|
|
"opt_einsum/_version.py",
|
|
"opt_einsum/backends/__init__.py",
|
|
"opt_einsum/backends/cupy.py",
|
|
"opt_einsum/backends/dispatch.py",
|
|
"opt_einsum/backends/tensorflow.py",
|
|
"opt_einsum/backends/theano.py",
|
|
"opt_einsum/backends/torch.py",
|
|
"opt_einsum/blas.py",
|
|
"opt_einsum/compat.py",
|
|
"opt_einsum/contract.py",
|
|
"opt_einsum/helpers.py",
|
|
"opt_einsum/parser.py",
|
|
"opt_einsum/paths.py",
|
|
"opt_einsum/sharing.py",
|
|
],
|
|
srcs_version = "PY3",
|
|
visibility = ["//visibility:public"],
|
|
)
|