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
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