Searched refs:find_cuda_config (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/third_party/nccl/ |
| D | nccl_configure.bzl | 17 "find_cuda_config",
71 …path = repository_ctx.path(Label("@org_tensorflow//third_party/gpus:find_cuda_config.py.gz.base64"…
77 cuda_config = find_cuda_config(repository_ctx, find_cuda_config_path, ["cuda"])
91 config = find_cuda_config(repository_ctx, find_cuda_config_path, ["nccl"])
|
| /external/tensorflow/third_party/tensorrt/ |
| D | tensorrt_configure.bzl | 11 "find_cuda_config",
104 …path = repository_ctx.path(Label("@org_tensorflow//third_party/gpus:find_cuda_config.py.gz.base64"…
112 config = find_cuda_config(repository_ctx, find_cuda_config_path, ["tensorrt"])
|
| /external/tensorflow/third_party/gpus/ |
| D | find_cuda_config.py | 572 def find_cuda_config(): function
645 for key, value in sorted(find_cuda_config().items()):
|
| D | cuda_configure.bzl | 621 # this case the trick is not sufficient as the find_cuda_config.py
641 def find_cuda_config(repository_ctx, script_path, cuda_libraries):
642 """Returns CUDA config dictionary from running find_cuda_config.py"""
645 auto_configure_fail("Failed to run find_cuda_config.py: %s" % err_out(exec_result))
666 config = find_cuda_config(repository_ctx, find_cuda_config_script, ["cuda", "cudnn"])
975 …ript = repository_ctx.path(Label("@org_tensorflow//third_party/gpus:find_cuda_config.py.gz.base64"…
|
| D | rocm_configure.bzl | 347 # this case the trick is not sufficient as the find_cuda_config.py
|
| /external/tensorflow/tensorflow/ |
| D | opensource_only.files | 125 tensorflow/third_party/gpus/find_cuda_config.py
126 tensorflow/third_party/gpus/find_cuda_config.py.gz.base64
|