Function _tpu_system_device_name

tensorflow/python/tpu/tpu.py:82–87

Returns the device name for the TPU_SYSTEM device of `job`.

(job)

def _tpu_system_device_name(job):
  """Returns the device name for the TPU_SYSTEM device of `job`."""
  if job is None:
    return "/device:TPU_SYSTEM:0"
  else:
    return "/job:%s/device:TPU_SYSTEM:0" % job


@tf_export(v1=["tpu.initialize_system"])