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