code: def testDatasetMap(self):
      # When a GPU is available, this would test that the wrapped call ops are
      # placed on the CPU (i.e. the device is selected using the unwrapped op).
      dataset_ops.Dataset.range(2).map(math_ops.square)
  
imports: import time
  from tensorflow.python.data.experimental.ops import prefetching_ops
  from tensorflow.python.data.ops import dataset_ops
  from tensorflow.python.eager import benchmarks_test_base
  from tensorflow.python.eager import context
  from tensorflow.python.eager import test
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import bitwise_ops
  from tensorflow.python.ops import critical_section_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import random_ops
  from tensorflow.python.ops.ragged import ragged_factory_ops
  from tensorflow.python.ops.ragged import ragged_map_ops
  from tensorflow.python.ops.ragged import ragged_tensor
  from tensorflow.python.util import tf_inspect
label_desc: Use image node_id 4 for calling the RunEagerOpAsFunctionTest obj's underlying member method code with example usage: obj.testDatasetMap() without return types