Searched refs:min_cuda_compute_capability (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/ |
| D | devices.cc | 31 const std::pair<int, int>& min_cuda_compute_capability) { in GetNumAvailableGPUs() argument
35 if (min_cuda_compute_capability.first != 0 || in GetNumAvailableGPUs()
36 min_cuda_compute_capability.second != 0) { in GetNumAvailableGPUs()
58 cuda_compute_capability >= min_cuda_compute_capability) { in GetNumAvailableGPUs()
71 << min_cuda_compute_capability.first << "." in GetNumAvailableGPUs()
72 << min_cuda_compute_capability.second << "): " << num_eligible_gpus; in GetNumAvailableGPUs()
80 << min_cuda_compute_capability.first << "." in GetNumAvailableGPUs()
81 << min_cuda_compute_capability.second << "): " << num_eligible_gpus in GetNumAvailableGPUs()
|
| D | devices.h | 33 const std::pair<int, int>& min_cuda_compute_capability = {0, 0});
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.test.pbtxt | 49 …argspec: "args=[\'cuda_only\', \'min_cuda_compute_capability\'], varargs=None, keywords=None, defa…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.test.pbtxt | 65 …argspec: "args=[\'cuda_only\', \'min_cuda_compute_capability\'], varargs=None, keywords=None, defa…
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | test_util.py | 1547 def is_gpu_available(cuda_only=False, min_cuda_compute_capability=None): argument
1591 if not min_cuda_compute_capability or cc >= min_cuda_compute_capability:
|
| D | config_test.py | 812 cuda_only=True, min_cuda_compute_capability=(8, 0)):
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | pooling_ops_test.py | 77 cuda_only=True, min_cuda_compute_capability=(6, 1)):
|