Searched refs:cuda_version (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/third_party/gpus/ |
| D | find_cuda_config.py | 153 def _get_default_cuda_paths(cuda_version): argument
154 if not cuda_version:
155 cuda_version = "*"
156 elif not "." in cuda_version:
157 cuda_version = cuda_version + ".*"
164 cuda_version)
166 return ["/usr/local/cuda-%s" % cuda_version, "/usr/local/cuda", "/usr",
258 cuda_version = header_version # x.y, see above.
260 cuda_library_path = _find_library(base_paths, "cudart", cuda_version)
275 ], nvcc_name, cuda_version, get_nvcc_version)
[all …]
|
| D | cuda_configure.bzl | 593 cuda_config.cuda_version,
660 cuda_version: The version of CUDA on the system.
671 cuda_version = config["cuda_version"].split(".")
672 cuda_major = cuda_version[0]
673 cuda_minor = cuda_version[1]
675 cuda_version = ("64_%s%s" if is_windows else "%s.%s") % (cuda_major, cuda_minor)
693 cudart_version = cuda_version
700 cudart_version = cuda_version
701 cublas_version = cuda_version
702 cusolver_version = cuda_version
[all …]
|
| /external/tensorflow/third_party/toolchains/preconfig/generate/ |
| D | BUILD | 35 cuda_version = "10.0",
49 cuda_version = "10.1",
61 cuda_version = "10.0",
71 cuda_version = "10.0",
97 cuda_version = "10.0",
110 cuda_version = "10.1",
121 cuda_version = "10.1",
|
| D | generate.bzl | 6 def _tensorflow_rbe_config(name, compiler, python_version, os, rocm_version = None, cuda_version = …
28 if cuda_version != None and rocm_version != None:
29 fail("Specifying both cuda_version and rocm_version is not supported.")
31 if cuda_version != None:
32 base = "@cuda%s-cudnn%s-%s//image" % (cuda_version, cudnn_version, os)
47 "TF_CUDA_VERSION": cuda_version,
|
| /external/tensorflow/third_party/toolchains/remote_config/ |
| D | configs.bzl | 21 cuda_version = "10.0",
33 cuda_version = "10.1",
46 cuda_version = "10.1",
59 cuda_version = "11.0",
75 cuda_version = "10.1",
86 cuda_version = "11.0",
|
| /external/tensorflow/third_party/nccl/ |
| D | nccl_configure.bzl | 78 cuda_version = cuda_config["cuda_version"].split(".")
87 {"%{cuda_version}": "(%s, %s)" % tuple(cuda_version)},
|
| D | build_defs.bzl.tpl | 7 _cuda_version = %{cuda_version}
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | auto_mixed_precision_lists.h | 107 AutoMixedPrecisionListsCuda(int cuda_version, int cudnn_version) in AutoMixedPrecisionListsCuda() argument
108 : cuda_version_(cuda_version), cudnn_version_(cudnn_version) {} in AutoMixedPrecisionListsCuda()
|
| /external/tensorflow/third_party/gpus/cuda/ |
| D | cuda_config.h.tpl | 19 #define TF_CUDA_VERSION "%{cuda_version}"
|
| /external/tensorflow/tensorflow/ |
| D | tensorflow.bzl | 2607 "cudart_dll_name": "cudart{cuda_version}.dll",
|