Function _tpu_init_fn

tensorflow/python/tpu/tpu_strategy_util.py:84–85

()

    @function.defun
    def _tpu_init_fn():
      return tpu.initialize_system(job=job)

    # The TPU_SYSTEM device must match the device used in tpu.initialize_system
    # exactly, otherwise you can get errors if there are multiple TPU_SYSTEM