Searched defs:cudnn_version (Results 1 – 8 of 8) sorted by relevance
56 tensorflow::CudnnVersion cudnn_version; in TEST_F() local74 tensorflow::CudnnVersion cudnn_version; in TEST_F() local
26 tensorflow.CudnnVersion cudnn_version = 3; field
53 tensorflow::ComputeCapability cc, tensorflow::CudnnVersion cudnn_version,
220 tensorflow::CudnnVersion cudnn_version; in GetCudnnVersion() local
256 se::port::StatusOr<se::dnn::VersionInfo> cudnn_version = in IsConvBF16Supported() local
109 AutoMixedPrecisionListsCuda(int cuda_version, int cudnn_version) in AutoMixedPrecisionListsCuda()
97 CudnnVersion cudnn_version = 3; field
112 tensorflow::CudnnVersion cudnn_version; in GetCudnnVersion() local