STT-tensorflow/tensorflow/python/distribute/experimental/BUILD

21 lines
530 B
Python

package(
default_visibility = ["//tensorflow:internal"],
licenses = ["notice"], # Apache 2.0
)
exports_files(["LICENSE"])
py_library(
name = "experimental",
srcs = [
"__init__.py",
],
srcs_version = "PY2AND3",
deps = [
"//tensorflow/python/distribute:central_storage_strategy",
"//tensorflow/python/distribute:collective_all_reduce_strategy",
"//tensorflow/python/distribute:parameter_server_strategy",
"//tensorflow/python/distribute:tpu_strategy",
],
)