Searched refs:prefetch_to_device (Results 1 – 8 of 8) sorted by relevance
39 prefetching_ops.prefetch_to_device("/cpu:1"))64 prefetching_ops.prefetch_to_device(89 prefetching_ops.prefetch_to_device("/cpu:1"))118 prefetching_ops.prefetch_to_device("/cpu:1"))148 prefetching_ops.prefetch_to_device("/gpu:0"))165 prefetching_ops.prefetch_to_device("/cpu:1"))197 prefetching_ops.prefetch_to_device("/gpu:0"))
26 def prefetch_to_device(device, buffer_size=None): function41 return prefetching_ops.prefetch_to_device(device, buffer_size)
101 from tensorflow.contrib.data.python.ops.prefetching_ops import prefetch_to_device
108 from tensorflow.python.data.experimental.ops.prefetching_ops import prefetch_to_device
36 def prefetch_to_device(device, buffer_size=None): function
164 name: "prefetch_to_device"
168 name: "prefetch_to_device"
481 * Add `tf.contrib.data.prefetch_to_device()`, which supports prefetching to GPU memory.489 …* Add `tf.contrib.data.prefetch_to_device`, which enables prefetching dataset elements to GPU memo…