Searched refs:_enclosing_tpu_context (Results 1 – 3 of 3) sorted by relevance
47 def _enclosing_tpu_context(): function79 tpu_context = _enclosing_tpu_context()147 if _enclosing_tpu_context() is None:176 if _enclosing_tpu_context() is None:232 if _enclosing_tpu_context() is None:
442 if values._enclosing_tpu_context() is not None: # pylint: disable=protected-access481 if values._enclosing_tpu_context() is not None: # pylint: disable=protected-access514 if values._enclosing_tpu_context() is not None:
755 def _enclosing_tpu_context(): function910 tpu_context = _enclosing_tpu_context()937 if _enclosing_tpu_context() is not None:1102 if _enclosing_tpu_context() is None:1149 if _enclosing_tpu_context() is None: