Home
last modified time | relevance | path

Searched refs:_thread_local_data (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/keras/
Dtesting_utils.py211 if _thread_local_data.run_eagerly is not None:
264 _thread_local_data = threading.local() variable
265 _thread_local_data.model_type = None
266 _thread_local_data.run_eagerly = None
267 _thread_local_data.experimental_run_tf_function = None
268 _thread_local_data.saved_model_format = None
283 previous_value = _thread_local_data.model_type
285 _thread_local_data.model_type = value
289 _thread_local_data.model_type = previous_value
305 previous_value = _thread_local_data.run_eagerly
[all …]
/external/tensorflow/tensorflow/python/eager/
Dcontext.py393 self._thread_local_data = _ThreadLocalData()
715 ctx = self._thread_local_data
732 return self._thread_local_data.is_eager
745 return self._thread_local_data.scope_name
750 self._thread_local_data.scope_name = s
755 return self._thread_local_data.device_name
760 return self._thread_local_data.device_spec
763 self._thread_local_data.device_name = device_name
764 self._thread_local_data.device_spec = device_spec
818 self._thread_local_data.executor = executor_new
[all …]
/external/tensorflow/tensorflow/python/distribute/
Dmirrored_strategy.py1013 eager_context_state = ctx._thread_local_data # pylint: disable=protected-access
1019 eager_context_state = ctx._thread_local_data # pylint: disable=protected-access
/external/tensorflow/tensorflow/python/framework/
Dops.py5659 c._thread_local_data.is_eager = False # pylint: disable=protected-access
5728 context._context._thread_local_data.is_eager = True