Home
last modified time | relevance | path

Searched refs:session_lib (Results 1 – 25 of 31) sorted by relevance

12

/external/tensorflow/tensorflow/python/training/
Dbasic_session_run_hooks_test.py27 from tensorflow.python.client import session as session_lib unknown
156 with session_lib.Session() as sess:
180 with session_lib.Session() as sess:
207 with session_lib.Session() as sess:
247 with ops.Graph().as_default(), session_lib.Session() as sess:
299 with ops.Graph().as_default(), session_lib.Session() as sess:
305 with ops.Graph().as_default(), session_lib.Session() as sess:
312 with ops.Graph().as_default(), session_lib.Session() as sess:
358 with ops.Graph().as_default(), session_lib.Session() as sess:
366 with ops.Graph().as_default(), session_lib.Session() as sess:
[all …]
Dlocalhost_cluster_performance_test.py25 from tensorflow.python.client import session as session_lib unknown
41 worker_sessions = [session_lib.Session(w.target) for w in workers]
84 worker_sessions = [session_lib.Session(w.target) for w in workers]
Dmonitored_session_test.py32 from tensorflow.python.client import session as session_lib unknown
1268 with ops.Graph().as_default(), session_lib.Session() as sess:
1285 with ops.Graph().as_default(), session_lib.Session() as sess:
1310 with ops.Graph().as_default(), session_lib.Session() as sess:
1326 with ops.Graph().as_default(), session_lib.Session() as sess:
1344 with ops.Graph().as_default(), session_lib.Session() as sess:
1361 with ops.Graph().as_default(), session_lib.Session() as sess:
1383 with ops.Graph().as_default(), session_lib.Session() as sess:
1401 with ops.Graph().as_default(), session_lib.Session() as sess:
2312 self.assertTrue(isinstance(session.raw_session(), session_lib.Session))
Dcheckpoint_utils_test.py27 from tensorflow.python.client import session as session_lib unknown
446 with session_lib.Session(
/external/tensorflow/tensorflow/python/saved_model/
Dload_v1_in_v2_test.py25 from tensorflow.python.client import session as session_lib unknown
83 with session_lib.Session() as session:
136 with session_lib.Session() as session:
165 with session_lib.Session() as session:
226 with session_lib.Session() as session:
286 with session_lib.Session() as session:
311 with session_lib.Session() as session:
344 with session_lib.Session() as session:
365 with session_lib.Session() as session:
386 with session_lib.Session() as session:
[all …]
Dsave_test.py28 from tensorflow.python.client import session as session_lib unknown
84 with graph.as_default(), session_lib.Session() as session:
495 with ops.Graph().as_default(), session_lib.Session() as session:
664 with ops.Graph().as_default(), session_lib.Session() as session:
968 with ops.Graph().as_default(), session_lib.Session() as session:
/external/tensorflow/tensorflow/python/framework/
Dconvert_to_constants_test.py33 from tensorflow.python.client import session as session_lib unknown
331 with session_lib.Session() as session:
572 with session_lib.Session() as sess:
592 with session_lib.Session() as sess:
608 with session_lib.Session() as sess:
638 with session_lib.Session() as sess:
673 with session_lib.Session() as sess:
720 with session_lib.Session() as sess:
739 with session_lib.Session() as sess:
765 with session_lib.Session() as sess:
[all …]
/external/tensorflow/tensorflow/python/tpu/
Dtpu_system_metadata.py24 from tensorflow.python.client import session as session_lib unknown
83 …devices = [session_lib._DeviceAttributes(device_util.canonicalize(d.name), # pylint: disable=prot…
94 with session_lib.Session(
180 with session_lib.Session(
Dtpu_strategy_util.py22 from tensorflow.python.client import session as session_lib unknown
124 with session_lib.Session(config=session_config, target=master) as sess:
211 with session_lib.Session(config=session_config, target=master) as sess:
Dasync_checkpoint.py31 from tensorflow.python.client import session as session_lib unknown
106 def after_create_session(self, session: session_lib.Session, coord: Any):
142 def end(self, session: session_lib.Session):
Dsession_support.py28 from tensorflow.python.client import session as session_lib unknown
50 return session_lib.Session(
227 self._session = session_lib.Session(
/external/tensorflow/tensorflow/compiler/tests/
Dxla_device_gpu_test.py21 from tensorflow.python.client import session as session_lib unknown
40 with session_lib.Session() as sess:
Djit_test.py28 from tensorflow.python.client import session as session_lib unknown
105 with session_lib.Session(config=NoRewriteSessionConfig()) as sess:
139 with session_lib.Session() as sess:
257 with session_lib.Session() as sess:
447 with session_lib.Session(graph=g, config=cfg) as sess:
487 with session_lib.Session(config=config) as sess:
528 with session_lib.Session(config=NoRewriteSessionConfig()) as sess:
592 with session_lib.Session(config=NoRewriteSessionConfig()) as sess:
622 with session_lib.Session(config=NoRewriteSessionConfig()) as sess:
/external/tensorflow/tensorflow/python/ops/
Dsplit_benchmark.py24 from tensorflow.python.client import session as session_lib unknown
95 with session_lib.Session(graph=graph, config=config) as session:
Dconcat_benchmark.py26 from tensorflow.python.client import session as session_lib unknown
104 with session_lib.Session(graph=graph, config=config) as session:
Dtranspose_benchmark.py25 from tensorflow.python.client import session as session_lib unknown
81 with session_lib.Session(graph=graph) as session:
Dmatmul_benchmark.py26 from tensorflow.python.client import session as session_lib unknown
91 with session_lib.Session(graph=graph) as session:
Dconv2d_benchmark.py26 from tensorflow.python.client import session as session_lib unknown
139 with session_lib.Session(graph=graph, config=config) as session:
Dbatch_norm_benchmark.py25 from tensorflow.python.client import session as session_lib unknown
147 with session_lib.Session(graph=graph) as session:
/external/tensorflow/tensorflow/python/training/tracking/
Dutil_with_v1_optimizers_test.py24 from tensorflow.python.client import session as session_lib unknown
145 first_session = session_lib.Session(graph=first_graph)
162 with second_graph.as_default(), session_lib.Session(graph=second_graph):
/external/tensorflow/tensorflow/python/kernel_tests/
Dpy_func_test.py29 from tensorflow.python.client import session as session_lib unknown
359 with session_lib.Session():
369 with session_lib.Session():
559 session = session_lib.Session(worker.target)
Dtensor_array_ops_test.py24 from tensorflow.python.client import session as session_lib unknown
82 session_lib.Session.reset(cls._workers[0].target)
1626 session = session_lib.Session(self._workers[0].target)
1657 session = session_lib.Session(self._workers[0].target)
1689 session = session_lib.Session(self._workers[0].target)
1855 self.run_op_benchmark(session_lib.Session(), op)
1868 self.run_op_benchmark(session_lib.Session(), op)
1874 self.run_op_benchmark(session_lib.Session(), op)
/external/tensorflow/tensorflow/python/keras/saving/
Dsaving_utils_test.py28 from tensorflow.python.client import session as session_lib unknown
254 with graph.as_default(), session_lib.Session() as session:
/external/tensorflow/tensorflow/python/distribute/
Dstrategy_test_lib.py29 from tensorflow.python.client import session as session_lib unknown
588 with session_lib.Session(target=workers[1].target) as sess:
604 with session_lib.Session(target=workers[1].target) as sess:
Dsharded_variable_test.py23 from tensorflow.python.client import session as session_lib unknown
54 with graph.as_default(), session_lib.Session() as session:

12