Searched refs:copy_to_device (Results 1 – 9 of 9) sorted by relevance
43 prefetching_ops.copy_to_device("/cpu:1"))68 prefetching_ops.copy_to_device("/cpu:1"))92 prefetching_ops.copy_to_device("/cpu:0"))117 prefetching_ops.copy_to_device("/cpu:1")).prefetch(1)142 prefetching_ops.copy_to_device("/cpu:1"))167 prefetching_ops.copy_to_device("/cpu:1")).prefetch(1)198 prefetching_ops.copy_to_device("/cpu:1"))231 prefetching_ops.copy_to_device("/cpu:1")).prefetch(1)261 prefetching_ops.copy_to_device("/gpu:0"))282 prefetching_ops.copy_to_device("/gpu:0")).prefetch(1)[all …]
45 def copy_to_device(target_device, source_device="/cpu:0"): function56 return prefetching_ops.copy_to_device(target_device, source_device)
63 dataset = dataset.apply(prefetching_ops.copy_to_device(
53 copy_to_device(target_device=device)).prefetch(buffer_size)59 def copy_to_device(target_device, source_device="/cpu:0"): function
100 from tensorflow.contrib.data.python.ops.prefetching_ops import copy_to_device
107 from tensorflow.python.data.experimental.ops.prefetching_ops import copy_to_device
104 name: "copy_to_device"
381 ds = (ds.apply(prefetching_ops.copy_to_device("/gpu:0"))