Home
last modified time | relevance | path

Searched defs:cudnn_version (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhlo_algorithm_denylist_test.cc48 tensorflow::CudnnVersion cudnn_version; in TEST_F() local
66 tensorflow::CudnnVersion cudnn_version; in TEST_F() local
Dgpu_autotuning.proto26 tensorflow.CudnnVersion cudnn_version = 3; field
Dhlo_algorithm_denylist.cc45 tensorflow::ComputeCapability cc, tensorflow::CudnnVersion cudnn_version,
Dgpu_conv_algorithm_picker.cc165 tensorflow::CudnnVersion cudnn_version; in GetCudnnVersion() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision_lists.h107 AutoMixedPrecisionListsCuda(int cuda_version, int cudnn_version) in AutoMixedPrecisionListsCuda()
/external/tensorflow/tensorflow/core/protobuf/
Dautotuning.proto79 CudnnVersion cudnn_version = 3; field
/external/tensorflow/tensorflow/core/kernels/
Dgpu_utils.cc100 tensorflow::CudnnVersion cudnn_version; in GetCudnnVersion() local