Searched refs:_get_header_version (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/third_party/gpus/ |
D | find_rocm_config.py | 57 def _get_header_version(path, name): function 105 major = _get_header_version(version_file, "HIP_VERSION_MAJOR") 106 minor = _get_header_version(version_file, "HIP_VERSION_MINOR") 123 major = _get_header_version(version_file, "MIOPEN_VERSION_MAJOR") 124 minor = _get_header_version(version_file, "MIOPEN_VERSION_MINOR") 125 patch = _get_header_version(version_file, "MIOPEN_VERSION_PATCH") 155 major = _get_header_version(version_file, "ROCBLAS_VERSION_MAJOR") 156 minor = _get_header_version(version_file, "ROCBLAS_VERSION_MINOR") 157 patch = _get_header_version(version_file, "ROCBLAS_VERSION_PATCH") 177 version_number = _get_header_version(version_file, "ROCRAND_VERSION") [all …]
|
D | find_cuda_config.py | 121 def _get_header_version(path, name): function 250 version = int(_get_header_version(path, "CUDA_VERSION")) 318 _get_header_version(path, name) 351 _get_header_version(path, name) 381 _get_header_version(path, name) 411 _get_header_version(path, name) 438 _get_header_version(path, name) 463 _get_header_version(path, name) 491 _get_header_version(path, name) 513 _get_header_version(path, name)
|