code: def testPrefetchToDeviceGpu(self):
      if not test_util.is_gpu_available():
          self.skipTest("No GPU available")
  
      host_dataset = dataset_ops.Dataset.range(10)
      device_dataset = host_dataset.apply(
          prefetching_ops.prefetch_to_device("/gpu:0")
      )
  
      self.assertDatasetProduces(device_dataset, list(range(10)))
  
imports: from absl.testing import parameterized
  from tensorflow.core.protobuf import config_pb2
  from tensorflow.python.data.experimental.ops import prefetching_ops
  from tensorflow.python.data.kernel_tests import test_base
  from tensorflow.python.data.ops import dataset_ops
  from tensorflow.python.data.util import structure
  from tensorflow.python.framework import combinations
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import sparse_tensor
  from tensorflow.python.framework import test_util
  from tensorflow.python.platform import test
label_desc: Use image node_id 5 for calling the PrefetchToDeviceTest obj's underlying member method code with example usage: obj.testPrefetchToDeviceGpu() without return types