Searched refs:prefetch_to_device (Results 1 – 9 of 9) sorted by relevance
43 prefetching_ops.prefetch_to_device("/cpu:1"))67 prefetching_ops.prefetch_to_device(92 prefetching_ops.prefetch_to_device("/cpu:1"))120 prefetching_ops.prefetch_to_device("/cpu:1"))149 prefetching_ops.prefetch_to_device("/gpu:0"))160 dataset = dataset.apply(prefetching_ops.prefetch_to_device("/gpu:0"))168 prefetching_ops.prefetch_to_device("/cpu:1"))199 prefetching_ops.prefetch_to_device("/gpu:0"))222 prefetching_ops.prefetch_to_device("/gpu:0"))
160 dataset = dataset.apply(prefetching_ops.prefetch_to_device("/gpu:0"))180 dataset = dataset.apply(prefetching_ops.prefetch_to_device("/gpu:0"))211 dataset = dataset.apply(prefetching_ops.prefetch_to_device("/gpu:0"))
285 prefetch_to_device=[None, True]))286 def test_prefetch_to_device_dataset(self, prefetch_to_device): argument291 if prefetch_to_device is None:295 experimental_prefetch_to_device=prefetch_to_device)
774 prefetch_to_device=[None, True]))775 def test_prefetch_to_device_dataset(self, prefetch_to_device): argument781 if prefetch_to_device is None:785 experimental_prefetch_to_device=prefetch_to_device)
129 from tensorflow.python.data.experimental.ops.prefetching_ops import prefetch_to_device
37 def prefetch_to_device(device, buffer_size=None): function
180 name: "prefetch_to_device"
208 name: "prefetch_to_device"
3154 * Add `tf.contrib.data.prefetch_to_device()`, which supports prefetching to GPU memory.3162 …* Add `tf.contrib.data.prefetch_to_device`, which enables prefetching dataset elements to GPU memo…