Searched refs:cuda_version_ (Results 1 – 2 of 2) sorted by relevance
108 : cuda_version_(cuda_version), cudnn_version_(cudnn_version) {} in AutoMixedPrecisionListsCuda()132 if (cuda_version_ >= 9010) { in AllowList()345 int cuda_version_;
938 cuda_version_(GetCudaVersion(*cluster)), in AutoMixedPrecisionImpl()952 return std::make_unique<AutoMixedPrecisionListsCuda>(cuda_version_, in get_mixed_precision_lists()1004 int cuda_version_; member in tensorflow::grappler::__anon69ea7d4c0111::AutoMixedPrecisionImpl