'Args:
ds (DataFlow): input DataFlow.
gpus (list[int]): list of GPUs to use. Will also start this number of processes.'

def __init__(self, ds, gpus):

    self.gpus = gpus
    super(PrefetchOnGPUs, self).__init__(ds, len(gpus))
