Searched refs:_find_library (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/third_party/gpus/ |
| D | find_cuda_config.py | 212 def _find_library(base_paths, library_name, required_version): function
260 cuda_library_path = _find_library(base_paths, "cudart", cuda_version)
285 cupti_library_path = _find_library(base_paths, "cupti", required_version)
336 library_path = _find_library(base_paths, "cublas", cublas_version)
366 library_path = _find_library(base_paths, "cusolver", cusolver_version)
396 library_path = _find_library(base_paths, "curand", curand_version)
425 library_path = _find_library(base_paths, "cufft", cufft_version)
448 library_path = _find_library(base_paths, "cudnn", cudnn_version)
478 library_path = _find_library(base_paths, "cusparse", cusparse_version)
500 library_path = _find_library(base_paths, "nccl", nccl_version)
[all …]
|