WeHub Mirror
16d3f15320
ARM CI / build (3.10) (push) Has been cancelled
cffconvert / validate (push) Has been cancelled
Creates a GitHub Issue when a PR Rolled back via Commit to Master / create-issue-on-pr-rollback (push) Has been cancelled
Scorecards supply-chain security / Scorecards analysis (push) Has been cancelled
102 lines
4.1 KiB
Python
102 lines
4.1 KiB
Python
# Copyright 2026 The TensorFlow Authors. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
# ==============================================================================
|
|
|
|
"""TensorFlow workspace initialization. Consult the WORKSPACE on how to use it."""
|
|
|
|
load("//third_party:repo.bzl", "tf_http_archive", "tf_mirror_urls", "tf_vendored")
|
|
load("//third_party/tf_runtime:workspace.bzl", tf_runtime = "repo")
|
|
|
|
def workspace():
|
|
tf_vendored(name = "xla", path = "third_party/xla")
|
|
tf_vendored(name = "tsl", path = "third_party/xla/third_party/tsl")
|
|
|
|
tf_http_archive(
|
|
name = "io_bazel_rules_closure",
|
|
sha256 = "5b00383d08dd71f28503736db0500b6fb4dda47489ff5fc6bed42557c07c6ba9",
|
|
strip_prefix = "rules_closure-308b05b2419edb5c8ee0471b67a40403df940149",
|
|
urls = tf_mirror_urls(
|
|
"https://github.com/bazelbuild/rules_closure/archive/308b05b2419edb5c8ee0471b67a40403df940149.tar.gz", # 2019-06-13
|
|
),
|
|
)
|
|
|
|
tf_runtime()
|
|
|
|
# https://github.com/bazelbuild/bazel-skylib/releases
|
|
tf_http_archive(
|
|
name = "bazel_skylib",
|
|
sha256 = "bc283cdfcd526a52c3201279cda4bc298652efa898b10b4db0837dc51652756f",
|
|
urls = tf_mirror_urls(
|
|
"https://github.com/bazelbuild/bazel-skylib/releases/download/1.7.1/bazel-skylib-1.7.1.tar.gz",
|
|
),
|
|
)
|
|
|
|
tf_http_archive(
|
|
name = "rules_license",
|
|
urls = tf_mirror_urls(
|
|
"https://github.com/bazelbuild/rules_license/releases/download/0.0.7/rules_license-0.0.7.tar.gz",
|
|
),
|
|
sha256 = "4531deccb913639c30e5c7512a054d5d875698daeb75d8cf90f284375fe7c360",
|
|
)
|
|
|
|
tf_http_archive(
|
|
name = "rules_pkg",
|
|
urls = tf_mirror_urls(
|
|
"https://github.com/bazelbuild/rules_pkg/releases/download/0.7.1/rules_pkg-0.7.1.tar.gz",
|
|
),
|
|
sha256 = "451e08a4d78988c06fa3f9306ec813b836b1d076d0f055595444ba4ff22b867f",
|
|
)
|
|
|
|
tf_http_archive(
|
|
name = "bazel_features",
|
|
sha256 = "4fd9922d464686820ffd8fcefa28ccffa147f7cdc6b6ac0d8b07fde565c65d66",
|
|
strip_prefix = "bazel_features-1.25.0",
|
|
urls = tf_mirror_urls(
|
|
"https://github.com/bazel-contrib/bazel_features/releases/download/v1.25.0/bazel_features-v1.25.0.tar.gz",
|
|
),
|
|
)
|
|
|
|
# Toolchains for ML projects hermetic builds.
|
|
# Details: https://github.com/google-ml-infra/rules_ml_toolchain
|
|
tf_http_archive(
|
|
name = "rules_ml_toolchain",
|
|
sha256 = "3b05687842427041c65d1bc2f4aeda3a3079557120f5be8c34690087a88c5de5",
|
|
strip_prefix = "rules_ml_toolchain-2eddbc595cc0bbe650c2640204f66b14f015f1a8",
|
|
urls = tf_mirror_urls(
|
|
"https://github.com/google-ml-infra/rules_ml_toolchain/archive/2eddbc595cc0bbe650c2640204f66b14f015f1a8.tar.gz",
|
|
),
|
|
)
|
|
|
|
# Maven dependencies.
|
|
RULES_JVM_EXTERNAL_TAG = "4.3"
|
|
tf_http_archive(
|
|
name = "rules_jvm_external",
|
|
strip_prefix = "rules_jvm_external-%s" % RULES_JVM_EXTERNAL_TAG,
|
|
sha256 = "6274687f6fc5783b589f56a2f1ed60de3ce1f99bc4e8f9edef3de43bdf7c6e74",
|
|
urls = tf_mirror_urls("https://github.com/bazelbuild/rules_jvm_external/archive/%s.zip" % RULES_JVM_EXTERNAL_TAG),
|
|
)
|
|
|
|
# Platforms
|
|
tf_http_archive(
|
|
name = "platforms",
|
|
sha256 = "29742e87275809b5e598dc2f04d86960cc7a55b3067d97221c9abbc9926bff0f",
|
|
urls = tf_mirror_urls(
|
|
"https://github.com/bazelbuild/platforms/releases/download/0.0.11/platforms-0.0.11.tar.gz",
|
|
),
|
|
)
|
|
|
|
# Alias so it can be loaded without assigning to a different symbol to prevent
|
|
# shadowing previous loads and trigger a buildifier warning.
|
|
tf_workspace3 = workspace
|