85 lines
2.4 KiB
Python
85 lines
2.4 KiB
Python
load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
|
|
|
|
# ==========================================
|
|
# Start of TensorFlow and its dependencies
|
|
# ==========================================
|
|
|
|
# This version of TensorFlow is injected only to make sure we use the same dependencies as TensorFlow (protobuffer, grpc, absl).
|
|
# TensorFlow is not compiled.
|
|
|
|
# Note: The OPs dynamic library depends on symbols specific to the version of
|
|
# absl used by tensorflow.
|
|
http_archive(
|
|
name = "org_tensorflow",
|
|
strip_prefix = "tensorflow-2.16.1",
|
|
sha256 = "c729e56efc945c6df08efe5c9f5b8b89329c7c91b8f40ad2bb3e13900bd4876d",
|
|
urls = ["https://github.com/tensorflow/tensorflow/archive/v2.16.1.tar.gz"],
|
|
# Starting with TF 2.14, disable hermetic Python builds.
|
|
patch_args = ["-p1"],
|
|
patches = ["//third_party/tensorflow:tf.patch"],
|
|
)
|
|
|
|
# Inject tensorflow dependencies.
|
|
# TensorFlow cannot anymore be injected from a sub-module.
|
|
# Note: The order is important.
|
|
load("@org_tensorflow//tensorflow:workspace3.bzl", tf1 = "workspace")
|
|
|
|
tf1()
|
|
|
|
load("@org_tensorflow//tensorflow:workspace2.bzl", tf2 = "workspace")
|
|
|
|
tf2()
|
|
|
|
load("@org_tensorflow//tensorflow:workspace1.bzl", tf3 = "workspace")
|
|
|
|
tf3()
|
|
|
|
load("@org_tensorflow//tensorflow:workspace0.bzl", tf4 = "workspace")
|
|
|
|
tf4()
|
|
|
|
# Inject TensorFlow from the Pypi package.
|
|
load("//third_party/tensorflow_pypi:tf_configure.bzl", "tf_configure")
|
|
|
|
tf_configure(name = "tensorflow_pypi")
|
|
|
|
# ========================================
|
|
# End of TensorFlow and its dependencies
|
|
# ========================================
|
|
|
|
# Third party libraries
|
|
load("//third_party/absl_py:workspace.bzl", absl_py = "deps")
|
|
load("//third_party/absl:workspace.bzl", absl = "deps")
|
|
load("//third_party/benchmark:workspace.bzl", benchmark = "deps")
|
|
load("//third_party/gtest:workspace.bzl", gtest = "deps")
|
|
load("//third_party/protobuf:workspace.bzl", protobuf = "deps")
|
|
|
|
absl()
|
|
absl_py()
|
|
benchmark()
|
|
gtest()
|
|
protobuf()
|
|
|
|
# Yggdrasil Decision Forests
|
|
load("//third_party/yggdrasil_decision_forests:workspace.bzl", yggdrasil_decision_forests = "deps")
|
|
|
|
yggdrasil_decision_forests()
|
|
|
|
load("@ydf//yggdrasil_decision_forests:library.bzl", ydf_load_deps = "load_dependencies")
|
|
|
|
ydf_load_deps(
|
|
exclude_repo = [
|
|
"absl",
|
|
"protobuf",
|
|
"zlib",
|
|
"farmhash",
|
|
"grpc",
|
|
"eigen",
|
|
"pybind11",
|
|
"pybind11_abseil",
|
|
"pybind11_protobuf",
|
|
"tensorflow"
|
|
],
|
|
repo_name = "@ydf",
|
|
)
|