Home
last modified time | relevance | path

Searched refs:_thread_local (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dmulti_worker_test_base.py463 self._thread_local = threading.local()
464 self._thread_local.cached_session = None
513 if getattr(self._thread_local, 'cached_session', None) is None:
514 self._thread_local.cached_session = session.Session(
516 sess = self._thread_local.cached_session
606 self._thread_local = threading.local()
610 if not hasattr(self._thread_local, 'dict'):
611 self._thread_local.dict = dict()
613 return dict.get(self._thread_local.dict, key, default)
618 if not hasattr(self._thread_local, 'dict'):
[all …]
Ddistribute_coordinator.py37 _thread_local = threading.local() variable
395 if getattr(_thread_local, "server", None) is not None:
396 assert _thread_local.cluster_spec == cluster_spec
397 assert _thread_local.task_type == task_type
398 assert _thread_local.task_id == task_id
399 assert _thread_local.session_config_str == repr(session_config)
400 assert _thread_local.rpc_layer == rpc_layer
401 assert _thread_local.environment == environment
402 return _thread_local.server
405 _thread_local.server_started = True
[all …]
Ddistribute_coordinator_test.py876 distribute_coordinator._thread_local = threading.local() # pylint: disable=protected-access
/external/tensorflow/tensorflow/python/framework/
Dops.py2954 self._thread_local = threading.local()
3072 self._thread_local._variable_creator_stack = new # pylint: disable=protected-access
3076 if self._thread_local._variable_creator_stack is not new: # pylint: disable=protected-access
3079 self._thread_local._variable_creator_stack = old # pylint: disable=protected-access
3085 if not hasattr(self._thread_local, "_variable_creator_stack"):
3086 self._thread_local._variable_creator_stack = [] # pylint: disable=protected-access
3105 return self._thread_local._variable_creator_stack # pylint: disable=protected-access
3109 …self._thread_local._variable_creator_stack = variable_creator_stack # pylint: disable=protected-a…
4155 if not hasattr(self._thread_local, "_name_stack"):
4156 self._thread_local._name_stack = ""
[all …]
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_layer.py387 self._thread_local = threading.local()
1622 self._thread_local._eager_losses = []
1625 layer._thread_local._eager_losses = []
3038 if not hasattr(self._thread_local, '_eager_losses'):
3039 self._thread_local._eager_losses = []
3040 return self._thread_local._eager_losses
3044 self._thread_local._eager_losses = losses
Dbase_layer_v1.py188 self._thread_local = threading.local()