• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1"""TensorFlow workspace initialization. Consult the WORKSPACE on how to use it."""
2
3load("//third_party/android:android_configure.bzl", "android_configure")
4load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
5load("@com_github_grpc_grpc//bazel:grpc_deps.bzl", "grpc_deps")
6load("@io_bazel_rules_closure//closure:defs.bzl", "closure_repositories")
7load("@rules_cuda//cuda:dependencies.bzl", "rules_cuda_dependencies")
8load("@rules_pkg//:deps.bzl", "rules_pkg_dependencies")
9
10# buildifier: disable=unnamed-macro
11def workspace(with_rules_cc = True):
12    """Loads a set of TensorFlow dependencies. To be used in a WORKSPACE file.
13
14    Args:
15      with_rules_cc: whether to load and patch rules_cc repository.
16    """
17    native.register_toolchains("@local_config_python//:py_toolchain")
18    rules_cuda_dependencies(with_rules_cc)
19    rules_pkg_dependencies()
20
21    closure_repositories()
22
23    http_archive(
24        name = "bazel_toolchains",
25        sha256 = "540cc8fec2bf8ab64d16fb9a7018f25738a4a03434057ea01b5d34add446ffb1",
26        strip_prefix = "bazel-toolchains-ea243d43269df23de03a797cff2347e1fc3d02bb",
27        urls = [
28            "http://mirror.tensorflow.org/github.com/bazelbuild/bazel-toolchains/archive/ea243d43269df23de03a797cff2347e1fc3d02bb.tar.gz",
29            "https://github.com/bazelbuild/bazel-toolchains/archive/ea243d43269df23de03a797cff2347e1fc3d02bb.tar.gz",
30        ],
31    )
32
33    android_configure(name = "local_config_android")
34
35    grpc_deps()
36
37# Alias so it can be loaded without assigning to a different symbol to prevent
38# shadowing previous loads and trigger a buildifier warning.
39tf_workspace1 = workspace
40