Searched defs:cuda_version (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/third_party/gpus/ |
| D | find_cuda_config.py | 153 def _get_default_cuda_paths(cuda_version): argument
312 def _find_cublas_config(base_paths, required_version, cuda_version): argument
345 def _find_cusolver_config(base_paths, required_version, cuda_version): argument
375 def _find_curand_config(base_paths, required_version, cuda_version): argument
405 def _find_cufft_config(base_paths, required_version, cuda_version): argument
457 def _find_cusparse_config(base_paths, required_version, cuda_version): argument
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | auto_mixed_precision_lists.h | 107 AutoMixedPrecisionListsCuda(int cuda_version, int cudnn_version) in AutoMixedPrecisionListsCuda()
|