blob: 663709c0cc3557a0fae71ccafa354512d5169b4b [file] [log] [blame]
"""TensorFlow workspace initialization. Consult the WORKSPACE on how to use it."""
load("//third_party/android:android_configure.bzl", "android_configure")
load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
load("@com_github_grpc_grpc//bazel:grpc_deps.bzl", "grpc_deps")
load("@io_bazel_rules_closure//closure:defs.bzl", "closure_repositories")
load("@rules_cuda//cuda:dependencies.bzl", "rules_cuda_dependencies")
load("@rules_pkg//:deps.bzl", "rules_pkg_dependencies")
def workspace():
native.register_toolchains("@local_config_python//:py_toolchain")
rules_cuda_dependencies()
rules_pkg_dependencies()
closure_repositories()
http_archive(
name = "bazel_toolchains",
sha256 = "540cc8fec2bf8ab64d16fb9a7018f25738a4a03434057ea01b5d34add446ffb1",
strip_prefix = "bazel-toolchains-ea243d43269df23de03a797cff2347e1fc3d02bb",
urls = [
"http://mirror.tensorflow.org/github.com/bazelbuild/bazel-toolchains/archive/ea243d43269df23de03a797cff2347e1fc3d02bb.tar.gz",
"https://github.com/bazelbuild/bazel-toolchains/archive/ea243d43269df23de03a797cff2347e1fc3d02bb.tar.gz",
],
)
android_configure(name = "local_config_android")
grpc_deps()
# Alias so it can be loaded without assigning to a different symbol to prevent
# shadowing previous loads and trigger a buildifier warning.
tf_workspace1 = workspace