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.py108 getattr(session.BaseSession, "profile_context", None) is None)
117 getattr(session.BaseSession, "profile_context", None) is None)
/external/tensorflow/tensorflow/python/distribute/cluster_resolver/
Dcluster_resolver_test.py49 @mock.patch.object(session.BaseSession, "list_devices")
69 @mock.patch.object(session.BaseSession, "list_devices")
93 @mock.patch.object(session.BaseSession, "list_devices")
Dtpu_cluster_resolver_test.py623 @mock.patch.object(session.BaseSession, 'list_devices')
650 @mock.patch.object(session.BaseSession, 'list_devices')
Dtfconfig_cluster_resolver_test.py247 @mock.patch.object(session.BaseSession, 'list_devices')
/external/tensorflow/tensorflow/python/client/
Dsession.py618 class BaseSession(SessionInterface): class
1338 m = BaseSession._NODEDEF_NAME_RE.search(message)
1364 if len(self._dead_handles) == BaseSession._DEAD_HANDLES_THRESHOLD:
1472 return BaseSession._Callable(self, callable_options)
1476 class Session(BaseSession):
1645 class InteractiveSession(BaseSession):
/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/debug/wrappers/
Dframework.py161 _check_type(sess, (session.BaseSession, monitored_session.MonitoredSession))
366 _check_type(sess, (session.BaseSession, monitored_session.MonitoredSession))