code: def get_tpu_strategy():
      resolver = get_tpu_cluster_resolver()
      remote.connect_to_cluster(resolver)
      tpu_cluster_resolver.initialize_tpu_system(resolver)
      strategy = tpu_lib.TPUStrategyV2(resolver)
      return strategy
  
file_path: tensorflow-upstream/tensorflow/python/distribute/tpu_strategy_compilation_test.py
imports: from tensorflow.python.distribute import tpu_strategy
  from tensorflow.python.distribute.cluster_resolver import tpu_cluster_resolver
  from tensorflow.python.eager import def_function
  from tensorflow.python.eager import remote
  from tensorflow.python.eager import test
  from tensorflow.python.framework import constant_op
  from tensorflow.python.platform import flags
label_desc: Use image node_id 2 for calling a global function with example usage: get_tpu_strategy() and returns: strategy