| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | hlo_algorithm_denylist_test.cc | 48 tensorflow::CudnnVersion cudnn_version; in TEST_F() local
49 cudnn_version.set_major(7); in TEST_F()
50 cudnn_version.set_minor(6); in TEST_F()
51 cudnn_version.set_patch(2); in TEST_F()
53 cc, cudnn_version, /*blas_version=*/"9000", in TEST_F()
66 tensorflow::CudnnVersion cudnn_version; in TEST_F() local
67 cudnn_version.set_major(7); in TEST_F()
68 cudnn_version.set_minor(6); in TEST_F()
69 cudnn_version.set_minor(2); in TEST_F()
71 GetDisabledConvAlgorithms(cc, cudnn_version, "9000", R"(invalid hlo)"); in TEST_F()
|
| D | hlo_algorithm_denylist.cc | 31 cudnn_version { major: 7 minor: 6 patch: 4 }
38 cudnn_version { major: 7 minor: 6 patch: 4 }
|
| D | hlo_algorithm_denylist.h | 28 tensorflow::ComputeCapability cc, tensorflow::CudnnVersion cudnn_version,
|
| D | gpu_autotuning.proto | 26 tensorflow.CudnnVersion cudnn_version = 3; field
|
| D | gpu_conv_algorithm_picker.cc | 165 tensorflow::CudnnVersion cudnn_version; in GetCudnnVersion() local
170 cudnn_version.set_major(version.major_version()); in GetCudnnVersion()
171 cudnn_version.set_minor(version.minor_version()); in GetCudnnVersion()
172 cudnn_version.set_patch(version.patch()); in GetCudnnVersion()
175 return cudnn_version; in GetCudnnVersion()
|
| /external/tensorflow/third_party/toolchains/preconfig/generate/ |
| D | BUILD | 37 cudnn_version = "7",
51 cudnn_version = "7",
62 cudnn_version = "7",
72 cudnn_version = "7",
98 cudnn_version = "7",
111 cudnn_version = "7",
122 cudnn_version = "7",
|
| D | generate.bzl | 6 …mpiler, python_version, os, rocm_version = None, cuda_version = None, cudnn_version = None, tensor…
32 base = "@cuda%s-cudnn%s-%s//image" % (cuda_version, cudnn_version, os)
46 "TF_CUDNN_VERSION": cudnn_version,
|
| /external/tensorflow/third_party/toolchains/remote_config/ |
| D | configs.bzl | 22 cudnn_version = "7",
34 cudnn_version = "7",
47 cudnn_version = "7",
60 cudnn_version = "8",
76 cudnn_version = "7",
87 cudnn_version = "8",
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 100 tensorflow::CudnnVersion cudnn_version; in GetCudnnVersion() local
105 cudnn_version.set_major(version.major_version()); in GetCudnnVersion()
106 cudnn_version.set_minor(version.minor_version()); in GetCudnnVersion()
107 cudnn_version.set_patch(version.patch()); in GetCudnnVersion()
110 return cudnn_version; in GetCudnnVersion()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/data/ |
| D | hlo_algorithm_denylist.pbtxt | 4 cudnn_version: {major: 7, minor: 6, patch: 0}
12 cudnn_version: {major: 7, minor: 6, patch: 2}
|
| /external/tensorflow/third_party/gpus/ |
| D | find_cuda_config.py | 446 cudnn_version = header_version.split(".")[0]
448 library_path = _find_library(base_paths, "cudnn", cudnn_version)
451 "cudnn_version": cudnn_version,
623 cudnn_version = os.environ.get("TF_CUDNN_VERSION", "")
624 result.update(_find_cudnn_config(cudnn_paths, cudnn_version))
|
| D | cuda_configure.bzl | 586 cuda_config.cudnn_version,
662 cudnn_version: The version of cuDNN on the system.
676 cudnn_version = ("64_%s" if is_windows else "%s") % config["cudnn_version"]
717 cudnn_version = cudnn_version,
843 "%{cudnn_version}": "",
1103 if cuda_config.cudnn_version.rsplit("_", 1)[0] >= "8":
1112 "cudnn_version.h",
1306 "%{cudnn_version}": cuda_config.cudnn_version,
1318 "cudnn_version": cuda_config.cudnn_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/tensorflow/stream_executor/cuda/ |
| D | BUILD | 378 ":cudnn_version",
404 ":cudnn_version",
640 name = "cudnn_version",
641 srcs = ["cudnn_version.cc"],
642 hdrs = ["cudnn_version.h"],
652 ":cudnn_version",
|
| /external/tensorflow/third_party/gpus/cuda/ |
| D | cuda_config.h.tpl | 26 #define TF_CUDNN_VERSION "%{cudnn_version}"
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | autotuning.proto | 79 CudnnVersion cudnn_version = 3; field
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | auto_mixed_precision_test.py | 587 cudnn_version = tuple([int(x) for x in cudnn_version_str.split('.')])
588 if cudnn_version < (8,):
|
| /external/tensorflow/tensorflow/ |
| D | tensorflow.bzl | 2608 "cudnn_dll_name": "cudnn{cudnn_version}.dll",
|