Searched defs:force_gpu (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| H A D | linear_operator_circulant_test.py | 225 def _constrain_devices_and_set_default(self, sess, use_gpu, force_gpu): argument
612 def _constrain_devices_and_set_default(self, sess, use_gpu, force_gpu): argument
913 def _constrain_devices_and_set_default(self, sess, use_gpu, force_gpu): argument
|
| H A D | linear_operator_toeplitz_test.py | 43 def _constrain_devices_and_set_default(self, sess, use_gpu, force_gpu): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | test_util.py | 1920 def force_gpu(): function
2623 def session(self, graph=None, config=None, use_gpu=True, force_gpu=False): argument
2671 force_gpu=False): argument
2725 force_gpu=False): argument
3594 def _constrain_devices_and_set_default(self, sess, use_gpu, force_gpu): argument
3614 def _create_session(self, graph, config, force_gpu): argument
3656 force_gpu=False, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | image_grad_test_base.py | 138 force_gpu=False): argument
|