Python shutdown_distributed_tpu Examples

Example #1

File: tpu.py Project: SylChan/tensorflow

def shutdown_system(job=None):
  """Shuts down a running a distributed TPU system."""
  if job is None:
    device_name = "/device:TPU_SYSTEM:0"
  else:
    device_name = "/job:%s/device:TPU_SYSTEM:0" % job
  with ops.device(device_name):
    shutdown_distributed_tpu = tpu_ops.shutdown_distributed_tpu()
  return shutdown_distributed_tpu

Example #2

File: tpu.py Project: codemogroup/Interview-Bot

def shutdown_system(job=None):
    """Shuts down a running a distributed TPU system."""
    if job is None:
        device_name = "/device:TPU_SYSTEM:0"
    else:
        device_name = "/job:%s/device:TPU_SYSTEM:0" % job
    with ops.device(device_name):
        shutdown_distributed_tpu = tpu_ops.shutdown_distributed_tpu()
    return shutdown_distributed_tpu

Example #3

File: tpu.py Project: mengdaya/TensorFlow

def shutdown_system(job=None):
    """Shuts down a running a distributed TPU system."""
    with ops.device(_tpu_system_device_name(job)):
        shutdown_distributed_tpu = tpu_ops.shutdown_distributed_tpu()
    return shutdown_distributed_tpu

Example #4

File: tpu.py Project: AnishShah/tensorflow

def shutdown_system(job=None):
  """Shuts down a running a distributed TPU system."""
  with ops.device(_tpu_system_device_name(job)):
    shutdown_distributed_tpu = tpu_ops.shutdown_distributed_tpu()
  return shutdown_distributed_tpu