| /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
1769 with self._create_session(graph, config, force_gpu) as sess:
1770 with self._constrain_devices_and_set_default(sess, use_gpu, force_gpu):
1778 force_gpu=False): argument
1820 graph, config, force_gpu, crash_if_inconsistent_args=True)
1822 force_gpu) as cached:
1832 force_gpu=False): argument
1841 graph, config, force_gpu, crash_if_inconsistent_args=False)
1843 force_gpu) as cached:
[all …]
|
| D | test_util_test.py | 77 with self.cached_session(force_gpu=True) as sess2:
370 with self.test_session(force_gpu=True):
|
| /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
45 self, sess, use_gpu, force_gpu) as sess:
370 def _constrain_devices_and_set_default(self, sess, use_gpu, force_gpu): argument
373 self, sess, use_gpu, force_gpu) as sess:
631 def _constrain_devices_and_set_default(self, sess, use_gpu, force_gpu): argument
634 self, sess, use_gpu, force_gpu) as sess:
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | huge_slice_op_test.py | 32 with self.cached_session(force_gpu=test.is_gpu_available()):
|
| D | cudnn_determinism_test.py | 68 with self.cached_session(force_gpu=True):
|
| D | init_ops_test.py | 552 with self.session(graph=graph, force_gpu=self.force_gpu):
558 for self.force_gpu in self._gpu_modes():
568 for self.force_gpu in self._gpu_modes():
579 for self.force_gpu in self._gpu_modes():
589 for self.force_gpu in self._gpu_modes():
596 for self.force_gpu in self._gpu_modes():
|
| D | constant_op_test.py | 945 with self.session(force_gpu=test_util.is_gpu_available()):
957 with self.session(force_gpu=test_util.is_gpu_available()):
968 with self.session(force_gpu=test_util.is_gpu_available()):
978 with self.session(force_gpu=test_util.is_gpu_available()):
|
| D | slice_op_test.py | 55 with self.cached_session(force_gpu=test.is_gpu_available()):
91 with self.cached_session(force_gpu=test.is_gpu_available()):
|
| D | unstack_op_test.py | 64 with test_util.force_gpu():
|
| D | spacetodepth_op_test.py | 45 with test_util.force_gpu():
|
| D | array_ops_test.py | 590 with test_util.force_gpu():
1290 with test_util.force_gpu():
1292 with test_util.force_gpu():
1301 with test_util.force_gpu():
|
| D | scatter_ops_test.py | 356 with test_util.force_gpu():
|
| D | scatter_nd_ops_test.py | 375 with self.cached_session(force_gpu=True):
|
| /external/tensorflow/tensorflow/contrib/resampler/python/ops/ |
| D | resampler_ops_test.py | 140 with self.test_session(use_gpu=on_gpu, force_gpu=False) as sess:
172 with self.test_session(use_gpu=on_gpu, force_gpu=False):
|
| /external/tensorflow/tensorflow/python/client/ |
| D | timeline_test.py | 98 with self.session(force_gpu=True) as sess:
|
| /external/tensorflow/tensorflow/contrib/optimizer_v2/ |
| D | adam_test.py | 112 with self.cached_session(force_gpu=test.is_gpu_available()):
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | adamax_test.py | 132 with self.cached_session(force_gpu=test.is_gpu_available()):
|
| D | lazy_adam_optimizer_test.py | 110 with self.cached_session(force_gpu=test.is_gpu_available()):
|
| D | adam_gs_optimizer_test.py | 118 with self.cached_session(force_gpu=test.is_gpu_available()):
|
| D | lazy_adam_gs_optimizer_test.py | 115 with self.cached_session(force_gpu=test.is_gpu_available()):
|
| /external/tensorflow/tensorflow/python/training/ |
| D | adam_test.py | 116 with self.cached_session(force_gpu=test.is_gpu_available()):
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | adamax_test.py | 134 with self.cached_session(force_gpu=test.is_gpu_available()):
|
| D | adam_test.py | 161 with self.cached_session(force_gpu=test.is_gpu_available()):
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | tf_trt_integration_test_base.py | 278 graph=g, config=config, use_gpu=True, force_gpu=True) as sess:
|
| /external/tensorflow/tensorflow/python/kernel_tests/signal/ |
| D | fft_ops_test.py | 72 with self.cached_session(config=config, force_gpu=True):
|