Searched refs:test_session (Results 1 – 25 of 61) sorted by relevance
123
| /external/tensorflow/tensorflow/contrib/reduce_slice_ops/python/kernel_tests/ |
| D | reduce_slice_ops_test.py | 34 with self.test_session(use_gpu=True):
43 with self.test_session(use_gpu=True):
56 with self.test_session(use_gpu=True):
69 with self.test_session(use_gpu=True):
79 with self.test_session(use_gpu=True):
88 with self.test_session(use_gpu=True):
97 with self.test_session(use_gpu=True):
106 with self.test_session(use_gpu=True):
114 with self.test_session(use_gpu=True):
122 with self.test_session(use_gpu=True):
[all …]
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | variable_ops_test.py | 47 with self.test_session() as sess, self.test_scope():
61 with self.test_session() as sess, self.test_scope():
75 with self.test_session() as sess, self.test_scope():
86 with self.test_session() as sess, self.test_scope():
98 with self.test_session() as sess, self.test_scope():
112 with self.test_session() as sess, self.test_scope():
125 with self.test_session() as session, self.test_scope():
141 with self.test_session() as session:
169 with self.test_session() as session:
206 with self.test_session() as sess, self.test_scope():
[all …]
|
| D | xla_ops_test.py | 38 with self.test_session() as session:
313 with self.test_session() as session:
326 with self.test_session() as session:
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/ |
| D | mpi_ops_test.py | 73 with self.test_session() as session:
80 with self.test_session() as session:
86 with self.test_session() as session:
125 with self.test_session(config=cpu_config) as session:
130 with self.test_session(config=gpu_config) as session:
160 with self.test_session() as session:
188 with self.test_session() as session:
204 with self.test_session() as session:
228 with self.test_session() as session:
263 with self.test_session() as session:
[all …]
|
| /external/tensorflow/tensorflow/lite/experimental/microfrontend/python/kernel_tests/ |
| D | audio_microfrontend_op_test.py | 39 with self.test_session():
58 with self.test_session():
80 with self.test_session():
100 with self.test_session():
123 with self.test_session():
143 with self.test_session():
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | multi_device_iterator_test.py | 49 with self.test_session(config=config):
60 with self.test_session(config=config):
79 with self.test_session(config=config):
98 with self.test_session(config=config):
122 with self.test_session(config=config):
150 with self.test_session(config=config) as sess:
179 with self.test_session(config=config):
208 with self.test_session(config=config) as sess:
240 with self.test_session(config=config):
261 with self.test_session(config=config):
[all …]
|
| /external/tensorflow/tensorflow/contrib/gan/python/eval/python/ |
| D | summaries_test.py | 109 with self.test_session(use_gpu=True):
129 with self.test_session(use_gpu=True):
145 with self.test_session(use_gpu=True):
161 with self.test_session(use_gpu=True):
171 with self.test_session(use_gpu=True):
180 with self.test_session(use_gpu=True) as sess:
|
| /external/tensorflow/tensorflow/python/kernel_tests/random/ |
| D | multinomial_op_big_test.py | 37 with self.test_session(use_gpu=True) as sess:
55 with self.test_session(use_gpu=True) as sess:
75 with self.test_session(use_gpu=True) as sess:
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | copy_to_device_test.py | 58 with self.test_session(config=worker_config):
83 with self.test_session(config=worker_config):
107 with self.test_session(config=worker_config):
132 with self.test_session(config=worker_config):
157 with self.test_session(config=worker_config):
182 with self.test_session(config=worker_config):
212 with self.test_session(config=worker_config):
245 with self.test_session(config=worker_config):
457 with self.test_session(config=worker_config):
486 with self.test_session(config=worker_config):
|
| D | prefetch_to_device_test.py | 54 with self.test_session(config=worker_config):
104 with self.test_session(config=worker_config):
132 with self.test_session(config=worker_config):
180 with self.test_session(config=worker_config):
|
| /external/tensorflow/tensorflow/contrib/memory_stats/python/kernel_tests/ |
| D | memory_stats_ops_test.py | 39 with self.test_session(use_gpu=True) as sess:
57 with self.test_session(use_gpu=True) as sess:
|
| /external/tensorflow/tensorflow/contrib/compiler/ |
| D | xla_test.py | 321 with self.test_session() as sess:
345 with self.test_session() as sess:
366 with self.test_session() as sess:
384 with self.test_session() as sess:
428 with self.test_session() as sess:
444 with self.test_session() as sess:
539 with self.test_session() as sess:
|
| /external/tensorflow/tensorflow/python/training/ |
| D | monitored_session_test.py | 912 with self.cached_session() as test_session:
913 session_creator = CountingSessionCreator(test_session)
938 with self.cached_session() as test_session:
939 session_creator = CountingSessionCreator(test_session)
968 with self.cached_session() as test_session:
969 session_creator = CountingSessionCreator(test_session)
993 with self.cached_session() as test_session:
994 session_creator = CountingSessionCreator(test_session)
1024 with self.cached_session() as test_session:
1025 session_creator = CountingSessionCreator(test_session)
[all …]
|
| /external/tensorflow/tensorflow/contrib/gan/python/ |
| D | train_test.py | 446 with self.test_session(use_gpu=True) as sess:
476 with self.test_session(use_gpu=True) as sess:
558 with self.test_session(use_gpu=True) as sess:
590 with self.test_session(use_gpu=True):
603 with self.test_session(use_gpu=True):
625 with self.test_session(use_gpu=True) as sess:
653 with self.test_session(use_gpu=True) as sess:
705 with self.test_session(use_gpu=True) as sess:
749 with self.test_session(use_gpu=True) as sess:
891 with self.test_session(use_gpu=True) as sess:
[all …]
|
| /external/tensorflow/tensorflow/contrib/nn/python/ops/ |
| D | scaled_softplus_test.py | 39 with self.test_session(use_gpu=True) as sess:
53 with self.test_session(use_gpu=True):
|
| /external/tensorflow/tensorflow/contrib/sparsemax/python/kernel_tests/ |
| D | sparsemax_loss_test.py | 82 with self.test_session(use_gpu=use_gpu):
92 with self.test_session(use_gpu=use_gpu):
207 with self.test_session(use_gpu=use_gpu):
224 with self.test_session(use_gpu=use_gpu):
|
| D | sparsemax_test.py | 73 with self.test_session(use_gpu=use_gpu):
227 with self.test_session(use_gpu=use_gpu):
241 with self.test_session(use_gpu=use_gpu):
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/state_space_models/ |
| D | test_utils.py | 46 with test_case.test_session():
83 with test_case.test_session():
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| D | cudnn_rnn_test.py | 317 with self.test_session(use_gpu=True) as sess:
368 with self.test_session(use_gpu=True, graph=g) as sess:
406 with self.test_session(use_gpu=True, graph=g) as sess:
549 with self.test_session(use_gpu=True, graph=g) as sess:
604 with self.test_session(use_gpu=True, graph=g) as sess:
652 with self.test_session(use_gpu=True, graph=g) as sess:
933 with self.test_session(use_gpu=True, graph=g) as sess:
969 with self.test_session(use_gpu=True, graph=g) as sess:
1010 with self.test_session(use_gpu=True, graph=g) as sess:
1052 with self.test_session(use_gpu=True, graph=ops.get_default_graph()):
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | functional_ops_test.py | 342 with self.test_session(use_gpu=True) as sess:
453 with self.test_session(config=worker_config) as sess:
582 with self.test_session(use_gpu=False) as sess:
832 with self.test_session(use_gpu=use_gpu) as sess:
856 with self.test_session(use_gpu=use_gpu):
908 with self.test_session(use_gpu=True):
933 with self.test_session(use_gpu=False) as sess:
981 with self.test_session(config=config):
1046 with self.test_session(config=config, use_gpu=True) as sess:
1092 with self.test_session(
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_grad_test.py | 57 with self.test_session():
108 with self.test_session():
|
| /external/tensorflow/tensorflow/contrib/image/python/kernel_tests/ |
| D | distort_image_ops_test.py | 75 with self.test_session(use_gpu=True):
135 with self.test_session(use_gpu=True):
192 with self.test_session(use_gpu=True):
|
| /external/tensorflow/tensorflow/contrib/remote_fused_graph/pylib/python/ops/ |
| D | remote_fused_graph_ops_test.py | 61 with self.test_session(use_gpu=False):
|
| /external/tensorflow/tensorflow/contrib/gan/python/losses/python/ |
| D | losses_impl_test.py | 616 with self.test_session(use_gpu=True):
641 with self.test_session(use_gpu=True):
656 with self.test_session(use_gpu=True):
667 with self.test_session(use_gpu=True) as sess:
695 with self.test_session(use_gpu=True):
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | test_util_test.py | 54 with self.test_session():
66 with self.test_session() as sess:
370 with self.test_session(force_gpu=True):
628 with self.test_session():
634 with self.test_session():
|
123