Searched refs:CudnnVersion (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cudnn_version_test.cc | 25 CudnnVersion version(7, 0, 12); in TEST()
33 /*source_version=*/CudnnVersion(7, 0, 12), in TEST()
34 /*loaded_version=*/CudnnVersion(7, 0, 14))); in TEST()
36 /*source_version=*/CudnnVersion(6, 1, 14), in TEST()
37 /*loaded_version=*/CudnnVersion(6, 1, 00))); in TEST()
42 /*source_version=*/CudnnVersion(7, 0, 12), in TEST()
43 /*loaded_version=*/CudnnVersion(6, 1, 14))); in TEST()
45 /*source_version=*/CudnnVersion(8, 1, 15), in TEST()
46 /*loaded_version=*/CudnnVersion(7, 0, 14))); in TEST()
51 /*source_version=*/CudnnVersion(7, 0, 14), in TEST()
[all …]
|
| D | cudnn_version.h | 26 struct CudnnVersion { struct
27 CudnnVersion() = default;
29 CudnnVersion(int major, int minor, int patch) in CudnnVersion() argument
43 bool IsSourceCompatibleWithCudnnLibrary(CudnnVersion source_version, argument
44 CudnnVersion loaded_version);
|
| D | cudnn_version.cc | 21 bool IsSourceCompatibleWithCudnnLibrary(CudnnVersion source_version, in IsSourceCompatibleWithCudnnLibrary()
22 CudnnVersion loaded_version) { in IsSourceCompatibleWithCudnnLibrary()
|
| D | cuda_dnn.cc | 287 port::Status GetLoadedCudnnVersion(CudnnVersion* version) { in GetLoadedCudnnVersion()
314 CudnnVersion source_version(CUDNN_MAJOR, CUDNN_MINOR, CUDNN_PATCHLEVEL); in Init()
316 CudnnVersion loaded_version; in Init()
359 CudnnVersion version; in GetVersion()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | hlo_algorithm_denylist_test.cc | 48 tensorflow::CudnnVersion cudnn_version; in TEST_F()
66 tensorflow::CudnnVersion cudnn_version; in TEST_F()
|
| D | hlo_algorithm_denylist.h | 28 tensorflow::ComputeCapability cc, tensorflow::CudnnVersion cudnn_version,
|
| D | gpu_autotuning.proto | 26 tensorflow.CudnnVersion cudnn_version = 3;
|
| D | gpu_conv_algorithm_picker.cc | 164 tensorflow::CudnnVersion GetCudnnVersion(se::StreamExecutor* stream_executor) { in GetCudnnVersion()
165 tensorflow::CudnnVersion cudnn_version; in GetCudnnVersion()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | autotuning.proto | 15 message CudnnVersion { message
79 CudnnVersion cudnn_version = 3;
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 99 tensorflow::CudnnVersion GetCudnnVersion(se::StreamExecutor* stream_executor) { in GetCudnnVersion()
100 tensorflow::CudnnVersion cudnn_version; in GetCudnnVersion()
|