code: def testTFRecordDataset(self):
      dataset = readers.TFRecordDataset(self._createTFRecordFiles())
      dataset = input_ops.auto_shard_dataset(
          dataset, self._num_shards, self._shard_index
      )
  
      self._verifySimpleShardingOutput(dataset, self._record)
  
imports: import os
  from tensorflow.python.data.ops import dataset_ops
  from tensorflow.python.data.ops import readers
  from tensorflow.python.data.util import structure
  from tensorflow.python.distribute import input_ops
  from tensorflow.python.eager import context
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import test_util
  from tensorflow.python.lib.io import python_io
  from tensorflow.python.ops import gen_dataset_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.platform import test
  from tensorflow.python.util import compat
label_desc: Use image node_id 10 for calling the AutoShardDatasetTest obj's underlying member method code with example usage: obj.testTFRecordDataset() without return types