Searched defs:cuda_version (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/third_party/gpus/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | auto_mixed_precision_lists.h | 109 AutoMixedPrecisionListsCuda(int cuda_version, int cudnn_version) in AutoMixedPrecisionListsCuda()
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/cuda/ |
| H A D | cutlass_utils.py | 120 cuda_version: Optional[str] = None variable in CUTLASSArgs
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | __init__.py | 154 from torch.version import cuda as cuda_version unknown
|