Searched refs:cudnn_version_str (Results 1 – 2 of 2) sorted by relevance
586 cudnn_version_str = sysconfig.get_build_info().get('cudnn_version', '0.0')587 cudnn_version = tuple([int(x) for x in cudnn_version_str.split('.')])
918 string cudnn_version_str = it->second; in GetCudnnVersion() local919 return std::stoi(cudnn_version_str); in GetCudnnVersion()