mirror of
https://github.com/tensorflow/tensorflow.git
synced 2024-11-21 21:05:19 +00:00
aa6da142f3
619575611 by A. Unique TensorFlower<gardener@tensorflow.org>:
Run buildifier on all files where it sorts loads differently
--
619498661 by A. Unique TensorFlower<gardener@tensorflow.org>:
[XLA:GPU][IndexAnalysis] Rename GetDefaultThreadIdToOutputIndexingMap to GetDefaultThreadIdIndexingMap.
The "output" part was a bit confusing. We use this function for threadId->input
mapping as well.
--
619490165 by A. Unique TensorFlower<gardener@tensorflow.org>:
Convert S8 to BF16 in one step without going though F32.
--
PiperOrigin-RevId: 619575611
34 lines
1.3 KiB
Python
34 lines
1.3 KiB
Python
"""TensorFlow workspace initialization. Consult the WORKSPACE on how to use it."""
|
|
|
|
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("@local_tsl//:workspace1.bzl", "tsl_workspace1")
|
|
load("@rules_pkg//:deps.bzl", "rules_pkg_dependencies")
|
|
|
|
# buildifier: disable=unnamed-macro
|
|
def workspace():
|
|
"""Loads a set of TensorFlow dependencies in a WORKSPACE file."""
|
|
tsl_workspace1()
|
|
|
|
native.register_toolchains("@local_config_python//:py_toolchain")
|
|
rules_pkg_dependencies()
|
|
|
|
closure_repositories()
|
|
|
|
http_archive(
|
|
name = "bazel_toolchains",
|
|
sha256 = "294cdd859e57fcaf101d4301978c408c88683fbc46fbc1a3829da92afbea55fb",
|
|
strip_prefix = "bazel-toolchains-8c717f8258cd5f6c7a45b97d974292755852b658",
|
|
urls = [
|
|
"http://mirror.tensorflow.org/github.com/bazelbuild/bazel-toolchains/archive/8c717f8258cd5f6c7a45b97d974292755852b658.tar.gz",
|
|
"https://github.com/bazelbuild/bazel-toolchains/archive/8c717f8258cd5f6c7a45b97d974292755852b658.tar.gz",
|
|
],
|
|
)
|
|
|
|
grpc_deps()
|
|
|
|
# Alias so it can be loaded without assigning to a different symbol to prevent
|
|
# shadowing previous loads and trigger a buildifier warning.
|
|
xla_workspace1 = workspace
|