A transformation that prefetches dataset values to the given <code>device</code>

A transformation that prefetches dataset values to the given device

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