Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision.cc49 const std::pair<int, int> kMinGPUArch = {7, 0}; variable
51 const std::pair<int, int> kMinGPUArch = {0, 0};
1163 return GetDeviceGPUArch(virtual_placer_.get_device(node)) >= kMinGPUArch; in IsOnSuitableGPUArch()
2005 : GetNumGPUs(*cluster, kMinGPUArch); in Optimize()
Dauto_mixed_precision_test.cc95 const std::pair<int, int> kMinGPUArch = {7, 0}; variable
105 gpu_available_ && (num_gpus == GetNumAvailableGPUs(kMinGPUArch)); in SetUp()