Searched refs:copy_to_device (Results 1 – 6 of 6) sorted by relevance
46 prefetching_ops.copy_to_device("/cpu:1"))71 prefetching_ops.copy_to_device("/cpu:1"))95 prefetching_ops.copy_to_device("/cpu:0"))120 prefetching_ops.copy_to_device("/cpu:1")).prefetch(1)145 prefetching_ops.copy_to_device("/cpu:1"))170 prefetching_ops.copy_to_device("/cpu:1")).prefetch(1)201 prefetching_ops.copy_to_device("/cpu:1"))234 prefetching_ops.copy_to_device("/cpu:1")).prefetch(1)264 prefetching_ops.copy_to_device("/gpu:0"))276 prefetching_ops.copy_to_device("/gpu:0")).prefetch(1)[all …]
54 copy_to_device(target_device=device)).prefetch(buffer_size)60 def copy_to_device(target_device, source_device="/cpu:0"): function
128 from tensorflow.python.data.experimental.ops.prefetching_ops import copy_to_device
104 name: "copy_to_device"
132 name: "copy_to_device"
388 ds = (ds.apply(prefetching_ops.copy_to_device("/gpu:0"))