code: def get_tpu_context():
      return _current_tpu_context
  
file_path: tensorflow/tensorflow/python/tpu/tpu_function.py
imports: import contextlib
  import threading
label_desc: Use image node_id 2 for calling a global function with example usage: get_tpu_context() and returns: _current_tpu_context