| """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 = "35c46f2fab174b3e7b474599da4e31713cc9a515" |
| TFRT_SHA256 = "3250df282ebad8161e8ce54f31bcad64ffb95b35442508f267f21c434c5f8171" |
| |
| 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 = None, |
| ) |