Searched refs:min_cuda_compute_capability (Results 1 – 8 of 8) 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()
54 min_cuda_compute_capability.first, in GetNumAvailableGPUs()
55 min_cuda_compute_capability.second)) { in GetNumAvailableGPUs()
68 << min_cuda_compute_capability.first << "." in GetNumAvailableGPUs()
69 << min_cuda_compute_capability.second << "): " << num_eligible_gpus; in GetNumAvailableGPUs()
77 << min_cuda_compute_capability.first << "." in GetNumAvailableGPUs()
78 << 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/python/grappler/ |
| D | remapper_test.py | 203 cuda_only=True, min_cuda_compute_capability=(8, 0)):
246 cuda_only=True, min_cuda_compute_capability=(8, 0)):
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.test.pbtxt | 53 …argspec: "args=[\'cuda_only\', \'min_cuda_compute_capability\'], varargs=None, keywords=None, defa…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.test.pbtxt | 69 …argspec: "args=[\'cuda_only\', \'min_cuda_compute_capability\'], varargs=None, keywords=None, defa…
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | test_util.py | 1846 def is_gpu_available(cuda_only=False, min_cuda_compute_capability=None): argument
1890 if not min_cuda_compute_capability or cc >= min_cuda_compute_capability:
|
| D | config_test.py | 889 cuda_only=True, min_cuda_compute_capability=(8, 0)):
|
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | pooling_ops_test.py | 145 cuda_only=True, min_cuda_compute_capability=(6, 1)):
214 cuda_only=True, min_cuda_compute_capability=(6, 1)):
|