Home
last modified time | relevance | path

Searched refs:session_lib (Results 1 – 25 of 29) 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(
Dsession_manager_test.py23 from tensorflow.python.client import session as session_lib unknown
152 with session_lib.Session():
427 s = session_lib.Session(server.target, graph=graph)
/external/tensorflow/tensorflow/python/saved_model/
Dload_v1_in_v2_test.py25 from tensorflow.python.client import session as session_lib unknown
81 with session_lib.Session() as session:
134 with session_lib.Session() as session:
163 with session_lib.Session() as session:
224 with session_lib.Session() as session:
284 with session_lib.Session() as session:
309 with session_lib.Session() as session:
342 with session_lib.Session() as session:
363 with session_lib.Session() as session:
384 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:
476 with ops.Graph().as_default(), session_lib.Session() as session:
/external/tensorflow/tensorflow/python/tpu/
Dtpu_system_metadata.py24 from tensorflow.python.client import session as session_lib unknown
64 …devices = [session_lib._DeviceAttributes(device_util.canonicalize(d.name), # pylint: disable=prot…
75 with session_lib.Session(
161 with session_lib.Session(
Dtpu_strategy_util.py22 from tensorflow.python.client import session as session_lib unknown
117 with session_lib.Session(config=session_config, target=master) as sess:
198 with session_lib.Session(config=session_config, target=master) as sess:
Dsession_support.py28 from tensorflow.python.client import session as session_lib unknown
50 return session_lib.Session(
228 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
35 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/keras/saving/
Dsaving_utils_test.py28 from tensorflow.python.client import session as session_lib unknown
241 with graph.as_default(), session_lib.Session() as session:
/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)
1584 session = session_lib.Session(self._workers[0].target)
1615 session = session_lib.Session(self._workers[0].target)
1647 session = session_lib.Session(self._workers[0].target)
1813 self.run_op_benchmark(session_lib.Session(), op)
1826 self.run_op_benchmark(session_lib.Session(), op)
1832 self.run_op_benchmark(session_lib.Session(), op)
/external/tensorflow/tensorflow/python/distribute/
Dstrategy_test_lib.py28 from tensorflow.python.client import session as session_lib unknown
553 with session_lib.Session(target=workers[1].target) as sess:
569 with session_lib.Session(target=workers[1].target) as sess:
/external/tensorflow/tensorflow/python/framework/
Dconvert_to_constants_test.py26 from tensorflow.python.client import session as session_lib unknown
260 with session_lib.Session() as session:
/external/tensorflow/tensorflow/python/training/tracking/
Dutil_with_v1_optimizers_test.py25 from tensorflow.python.client import session as session_lib unknown
602 first_session = session_lib.Session(graph=first_graph)
619 with second_graph.as_default(), session_lib.Session(graph=second_graph):
Dutil.py28 from tensorflow.python.client import session as session_lib unknown
1032 class _SessionWithFeedDictAdditions(session_lib.SessionInterface):

12