Home
last modified time | relevance | path

Searched refs:GetCudaVersion (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/platform/default/
Ddso_loader.cc33 string GetCudaVersion() { return TF_CUDA_VERSION; } in GetCudaVersion() function
110 auto status_or_handle = GetDsoHandle("cupti", GetCudaVersion()); in GetCuptiDsoHandle()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision.cc894 int GetCudaVersion(const Cluster& cluster) { in GetCudaVersion() function
938 cuda_version_(GetCudaVersion(*cluster)), in AutoMixedPrecisionImpl()
Dauto_mixed_precision_test.cc1023 int GetCudaVersion(const Cluster& cluster) { in GetCudaVersion() function
1058 if (GetCudaVersion(*virtual_cluster_.get()) >= 9010) { in TEST_F()