25 lines
619 B
Python
25 lines
619 B
Python
# Description:
|
|
# Python APIs for various Tensorflow backends.
|
|
|
|
load("//tensorflow:tensorflow.bzl", "if_windows")
|
|
|
|
package(
|
|
default_visibility = ["//visibility:public"],
|
|
licenses = ["notice"], # Apache 2.0
|
|
)
|
|
|
|
exports_files(["LICENSE"])
|
|
|
|
py_library(
|
|
name = "compiler",
|
|
srcs = ["__init__.py"],
|
|
srcs_version = "PY3",
|
|
deps = if_windows(
|
|
["//tensorflow/python/compiler/tensorrt:trt_convert_windows"],
|
|
otherwise = ["//tensorflow/python/compiler/tensorrt:init_py"],
|
|
) + [
|
|
"//tensorflow/python/compiler/mlir",
|
|
"//tensorflow/python/compiler/xla:compiler_py",
|
|
],
|
|
)
|