77 lines
2.2 KiB
Python
77 lines
2.2 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.12.0",
|
|
sha256 = "f67d5fdc15c38a50eb95d8788125ed7758e946b6c7313c188fe1d2fc9a9dfa1e",
|
|
urls = ["https://github.com/tensorflow/tensorflow/archive/v2.12.0.zip"],
|
|
)
|
|
|
|
# Inject tensorflow dependencies.
|
|
# TensorFlow cannot anymore be injected from a sub-module.
|
|
# Note: The other 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",
|
|
],
|
|
repo_name = "@ydf",
|
|
)
|