| dataset_prefetch_to_device {tfdatasets} | R Documentation |
A transformation that prefetches dataset values to the given device
Description
A transformation that prefetches dataset values to the given device
Usage
dataset_prefetch_to_device(dataset, device, buffer_size = NULL)
Arguments
dataset |
A dataset |
device |
A string. The name of a device to which elements will be prefetched (e.g. "/gpu:0"). |
buffer_size |
(Optional.) The number of elements to buffer on device. Defaults to an automatically chosen value. |
Value
A dataset
Note
Although the transformation creates a dataset, the transformation must be the final dataset in the input pipeline.
See Also
Other dataset methods:
dataset_batch(),
dataset_cache(),
dataset_collect(),
dataset_concatenate(),
dataset_decode_delim(),
dataset_filter(),
dataset_interleave(),
dataset_map(),
dataset_map_and_batch(),
dataset_padded_batch(),
dataset_prefetch(),
dataset_reduce(),
dataset_repeat(),
dataset_shuffle(),
dataset_shuffle_and_repeat(),
dataset_skip(),
dataset_take(),
dataset_take_while(),
dataset_window()