mirror of
https://github.com/tensorflow/tensorflow.git
synced 2024-11-21 21:05:19 +00:00
e2137a01a5
PiperOrigin-RevId: 445232402
18 lines
854 B
Python
18 lines
854 B
Python
"""Loads Vulkan-Headers, used by TF Lite."""
|
|
|
|
load("//third_party:repo.bzl", "tf_http_archive", "tf_mirror_urls")
|
|
|
|
def repo():
|
|
tf_http_archive(
|
|
name = "vulkan_headers",
|
|
# LINT.IfChange
|
|
strip_prefix = "Vulkan-Headers-32c07c0c5334aea069e518206d75e002ccd85389",
|
|
sha256 = "602aedcc4c6057473d0f7fee1bcc3aa01bf191371b2b5bbca949cebc03cf393a",
|
|
link_files = {
|
|
"//third_party/vulkan_headers:tensorflow/vulkan_hpp_dispatch_loader_dynamic.cc": "tensorflow/vulkan_hpp_dispatch_loader_dynamic.cc",
|
|
},
|
|
urls = tf_mirror_urls("https://github.com/KhronosGroup/Vulkan-Headers/archive/32c07c0c5334aea069e518206d75e002ccd85389.tar.gz"),
|
|
# LINT.ThenChange(//tensorflow/lite/tools/cmake/modules/vulkan_headers.cmake)
|
|
build_file = "//third_party/vulkan_headers:vulkan_headers.BUILD",
|
|
)
|