wehub-resource-sync
21 lines
866 B
Python
21 lines
866 B
Python
"""Provides the repository macro to import TFRT."""
|
|
|
|
load("//third_party:repo.bzl", "tf_http_archive", "tf_mirror_urls")
|
|
|
|
def repo():
|
|
"""Imports TFRT."""
|
|
|
|
# Attention: tools parse and update these lines.
|
|
TFRT_COMMIT = "4ecc3a44a32c832b748328bed3f9a599f795ca8d"
|
|
TFRT_SHA256 = "5e81d70f9534340f7ef8e63ec43bdd5971135e48183079be50ecb3f74b1fed66"
|
|
|
|
tf_http_archive(
|
|
name = "tf_runtime",
|
|
sha256 = TFRT_SHA256,
|
|
strip_prefix = "runtime-{commit}".format(commit = TFRT_COMMIT),
|
|
urls = tf_mirror_urls("https://github.com/tensorflow/runtime/archive/{commit}.tar.gz".format(commit = TFRT_COMMIT)),
|
|
# A patch file can be provided for atomic commits to both TF and TFRT.
|
|
# The job that bumps the TFRT_COMMIT also resets patch_file to 'None'.
|
|
patch_file = ["//third_party/tf_runtime:f16_attr.patch"],
|
|
)
|