deviceA transformation that prefetches dataset values to the given device
dataset_prefetch_to_device(dataset, device, buffer_size = NULL)A dataset
A string. The name of a device to which elements will be prefetched (e.g. "/gpu:0").
(Optional.) The number of elements to buffer on device. Defaults to an automatically chosen value.
A dataset
Other dataset methods: dataset_batch,
dataset_cache,
dataset_concatenate,
dataset_decode_delim,
dataset_filter,
dataset_interleave,
dataset_map_and_batch,
dataset_map,
dataset_padded_batch,
dataset_prefetch,
dataset_repeat,
dataset_shuffle_and_repeat,
dataset_shuffle,
dataset_skip, dataset_take