https://github.com/google/jax
Raw File
Tip revision: 4198657c6758b40a493d52c0c47ef91f8a61d8c8 authored by Jake VanderPlas on 17 March 2021, 17:50:27 UTC
JIT: Convert python integers to uint64 when applicable rather than overflowing.
Tip revision: 4198657
WORKSPACE
load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")

# To update TensorFlow to a new revision,
# a) update URL and strip_prefix to the new git commit hash
# b) get the sha256 hash of the commit by running:
#    curl -L https://github.com/tensorflow/tensorflow/archive/<git hash>.tar.gz | sha256sum
#    and update the sha256 with the result.
http_archive(
    name = "org_tensorflow",
    sha256 = "8261fa982a1338e4dc0c3105a772231c6f5bf73064c046e28e4596d4b36895f1",
    strip_prefix = "tensorflow-fa534185655c95abe50e55d0125dd64c6b92ccfd",
    urls = [
        "https://github.com/tensorflow/tensorflow/archive/fa534185655c95abe50e55d0125dd64c6b92ccfd.tar.gz",
    ],
)

# For development, one can use a local TF repository instead.
# local_repository(
#    name = "org_tensorflow",
#    path = "tensorflow",
# )

load("//third_party/pocketfft:workspace.bzl", pocketfft = "repo")
pocketfft()

# Initialize TensorFlow's external dependencies.
load("@org_tensorflow//tensorflow:workspace3.bzl", "workspace")
workspace()
load("@org_tensorflow//tensorflow:workspace2.bzl", "workspace")
workspace()
load("@org_tensorflow//tensorflow:workspace1.bzl", "workspace")
workspace()
load("@org_tensorflow//tensorflow:workspace0.bzl", "workspace")
workspace()
back to top