| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | init_ops_test.py | 557 with self.session(graph=graph, force_gpu=self.force_gpu):
563 for self.force_gpu in self._gpu_modes():
573 for self.force_gpu in self._gpu_modes():
584 for self.force_gpu in self._gpu_modes():
594 for self.force_gpu in self._gpu_modes():
601 for self.force_gpu in self._gpu_modes():
627 with self.session(graph=graph, force_gpu=self.force_gpu):
634 with self.session(graph=graph, force_gpu=self.force_gpu):
645 with self.session(graph=graph, force_gpu=self.force_gpu) as sess:
650 for self.force_gpu in self._gpu_modes():
[all …]
|
| D | huge_slice_op_test.py | 32 with self.cached_session(force_gpu=test.is_gpu_available()):
|
| D | cudnn_deterministic_base.py | 86 with self.cached_session(force_gpu=True):
|
| D | bias_op_deterministic_test.py | 80 with self.session(force_gpu=True):
|
| D | constant_op_test.py | 989 with self.session(force_gpu=test_util.is_gpu_available()):
1001 with self.session(force_gpu=test_util.is_gpu_available()):
1012 with self.session(force_gpu=test_util.is_gpu_available()):
1022 with self.session(force_gpu=test_util.is_gpu_available()):
|
| D | scatter_ops_test.py | 327 with test_util.force_gpu():
|
| D | spacetodepth_op_test.py | 46 with test_util.force_gpu():
|
| D | array_ops_test.py | 673 with test_util.force_gpu():
1455 with test_util.force_gpu():
1457 with test_util.force_gpu():
1466 with test_util.force_gpu():
|
| D | control_flow_ops_py_test.py | 4592 with self.cached_session(force_gpu=test.is_gpu_available()) as sess:
4615 with self.cached_session(force_gpu=test.is_gpu_available()) as sess:
4637 with self.cached_session(force_gpu=test.is_gpu_available()) as sess:
|
| D | tensor_array_ops_test.py | 1782 with self.session(force_gpu=True) as sess:
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | test_util.py | 1621 def force_gpu(): function
2276 def session(self, graph=None, config=None, use_gpu=True, force_gpu=False): argument
2315 with self._create_session(graph, config, force_gpu) as sess:
2316 with self._constrain_devices_and_set_default(sess, use_gpu, force_gpu):
2324 force_gpu=False): argument
2366 graph, config, force_gpu, crash_if_inconsistent_args=True)
2368 force_gpu) as cached:
2378 force_gpu=False): argument
2391 graph, config, force_gpu, crash_if_inconsistent_args=False)
2393 force_gpu) as cached:
[all …]
|
| D | test_util_test.py | 81 with self.cached_session(force_gpu=True) as sess2:
394 with self.test_session(force_gpu=True):
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | linear_operator_toeplitz_test.py | 47 def _constrain_devices_and_set_default(self, sess, use_gpu, force_gpu): argument
50 self, sess, use_gpu, force_gpu) as sess:
|
| D | linear_operator_circulant_test.py | 58 def _constrain_devices_and_set_default(self, sess, use_gpu, force_gpu): argument
61 self, sess, use_gpu, force_gpu) as sess:
410 def _constrain_devices_and_set_default(self, sess, use_gpu, force_gpu): argument
413 self, sess, use_gpu, force_gpu) as sess:
703 def _constrain_devices_and_set_default(self, sess, use_gpu, force_gpu): argument
706 self, sess, use_gpu, force_gpu) as sess:
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | unstack_op_test.py | 92 with test_util.force_gpu():
211 with test_util.force_gpu():
|
| D | slice_op_test.py | 59 with self.cached_session(force_gpu=test.is_gpu_available()):
95 with self.cached_session(force_gpu=test.is_gpu_available()):
|
| D | scatter_nd_ops_test.py | 450 with self.cached_session(force_gpu=True):
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_grad_deterministic_test.py | 76 with self.session(force_gpu=True):
|
| D | image_grad_test_base.py | 140 force_gpu=False): argument
141 with self.cached_session(use_gpu=use_gpu, force_gpu=force_gpu):
|
| /external/tensorflow/tensorflow/python/client/ |
| D | timeline_test.py | 98 with self.session(force_gpu=True) as sess:
|
| /external/cpuinfo/test/build.prop/ |
| D | galaxy-s6.log | 117 debug.hwc.force_gpu=0
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | adamax_test.py | 132 force_gpu=test.is_gpu_available()):
|
| D | adam_test.py | 162 force_gpu=test.is_gpu_available()):
614 force_gpu=test.is_gpu_available()):
|
| /external/tensorflow/tensorflow/python/training/ |
| D | adam_test.py | 117 with self.cached_session(force_gpu=test.is_gpu_available()):
|
| /external/tensorflow/tensorflow/python/kernel_tests/signal/ |
| D | fft_ops_test.py | 82 with self.cached_session(config=config, force_gpu=True):
|