Python shutdown_distributed_tpuの例

コード例 #1

ファイル: tpu.py プロジェクト: 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

コード例 #2

ファイル: tpu.py プロジェクト: 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

コード例 #3

ファイル: tpu.py プロジェクト: 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

コード例 #4

ファイル: tpu.py プロジェクト: 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