Home
last modified time | relevance | path

Searched refs:BaseSession (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/profiler/
Dprofile_context.py331 self.old_run = getattr(session.BaseSession, 'run', None)
332 self.old_init = getattr(session.BaseSession, '__init__', None)
338 elif getattr(session.BaseSession, '_profiler_run_internal', None):
341 elif getattr(session.BaseSession, '_profiler_init_internal', None):
345 setattr(session.BaseSession, 'run', _profiled_run)
346 setattr(session.BaseSession, '__init__', _profiled_init)
347 setattr(session.BaseSession, '_profiler_run_internal', self.old_run)
348 setattr(session.BaseSession, '_profiler_init_internal', self.old_init)
349 setattr(session.BaseSession, 'profile_context', self)
358 setattr(session.BaseSession, 'run', self.old_run)
[all …]
Dprofile_context_test.py115 getattr(session.BaseSession, "profile_context", None) is None)
124 getattr(session.BaseSession, "profile_context", None) is None)
/external/tensorflow/tensorflow/python/distribute/cluster_resolver/
Dcluster_resolver_test.py50 @mock.patch.object(session.BaseSession, "list_devices")
70 @mock.patch.object(session.BaseSession, "list_devices")
94 @mock.patch.object(session.BaseSession, "list_devices")
Dtfconfig_cluster_resolver_test.py263 @mock.patch.object(session.BaseSession, 'list_devices')
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-interactive-session.pbtxt4 is_instance: "<class \'tensorflow.python.client.session.BaseSession\'>"
Dtensorflow.-session.pbtxt4 is_instance: "<class \'tensorflow.python.client.session.BaseSession\'>"
/external/tensorflow/tensorflow/python/client/
Dsession.py637 class BaseSession(SessionInterface): class
1378 m = BaseSession._NODEDEF_NAME_RE.search(message)
1409 if len(self._dead_handles) == BaseSession._DEAD_HANDLES_THRESHOLD:
1515 return BaseSession._Callable(self, callable_options)
1519 class Session(BaseSession):
1689 class InteractiveSession(BaseSession):
/external/tensorflow/tensorflow/python/distribute/cluster_resolver/tpu/
Dtpu_cluster_resolver_test.py613 @mock.patch.object(session.BaseSession, 'list_devices')
668 @mock.patch.object(session.BaseSession, 'list_devices')
/external/tensorflow/tensorflow/python/debug/wrappers/
Dframework.py148 _check_type(sess, (session.BaseSession, monitored_session.MonitoredSession))
346 _check_type(sess, (session.BaseSession, monitored_session.MonitoredSession))