Class SingleDevice

tensorflow/python/distribute/numpy_dataset.py:93–97

Used with `colocate_with` to create a non-mirrored variable.

class SingleDevice(object):
  """Used with `colocate_with` to create a non-mirrored variable."""

  def __init__(self, device):
    self.device = device