Searched refs:cudnn_version_ (Results 1 – 2 of 2) sorted by relevance
108 : cuda_version_(cuda_version), cudnn_version_(cudnn_version) {} in AutoMixedPrecisionListsCuda()137 if (cudnn_version_ >= 7602) { in AllowList()145 if (cudnn_version_ >= 8000) { in AllowList()346 int cudnn_version_; variable
939 cudnn_version_(GetCudnnVersion(*cluster)), in AutoMixedPrecisionImpl()953 cudnn_version_); in get_mixed_precision_lists()1005 int cudnn_version_; member in tensorflow::grappler::__anon69ea7d4c0111::AutoMixedPrecisionImpl