Searched defs:force_gpu (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | linear_operator_circulant_test.py | 42 def _constrain_devices_and_set_default(self, sess, use_gpu, force_gpu): argument
370 def _constrain_devices_and_set_default(self, sess, use_gpu, force_gpu): argument
631 def _constrain_devices_and_set_default(self, sess, use_gpu, force_gpu): argument
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | test_util.py | 1333 def force_gpu(): function
1731 def session(self, graph=None, config=None, use_gpu=False, force_gpu=False): argument
1778 force_gpu=False): argument
1832 force_gpu=False): argument
2538 def _constrain_devices_and_set_default(self, sess, use_gpu, force_gpu): argument
2558 def _create_session(self, graph, config, force_gpu): argument
2602 force_gpu=False, argument
|