tf.contrib.data.prefetch_to_device(
device,
buffer_size=None
)
Defined in tensorflow/contrib/data/python/ops/prefetching_ops.py
.
A transformation that prefetches dataset values to the given device
. (deprecated)
NOTE: Although the transformation creates a tf.data.Dataset
, the
transformation must be the final Dataset
in the input pipeline.
Args:
device
: A string. The name of a device to which elements will be prefetched.buffer_size
: (Optional.) The number of elements to buffer ondevice
. Defaults to an automatically chosen value.
Returns:
A Dataset
transformation function, which can be passed to
tf.data.Dataset.apply
.