Home
last modified time | relevance | path

Searched refs:gpu_available_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision_test.cc102 gpu_available_ = (num_gpus > 0); in SetUp()
104 gpu_available_ = in SetUp()
105 gpu_available_ && (num_gpus == GetNumAvailableGPUs(kMinGPUArch)); in SetUp()
107 if (gpu_available_) { in SetUp()
206 bool gpu_available_; member in tensorflow::grappler::__anone3282b120111::AutoMixedPrecisionTest
Dlayout_optimizer_test.cc44 gpu_available_ = GetNumAvailableGPUs() > 0; in SetUp()
46 if (gpu_available_) { in SetUp()
227 bool gpu_available_; member in tensorflow::grappler::__anonb8195e580111::LayoutOptimizerTest
252 if (gpu_available_) { in TEST_F()