code: def _shard(
      input_dataset, num_shards, index, name
  ):  # pylint: disable=unused-private-name
      """See `Dataset.shard()` for details."""
      return _ShardDataset(input_dataset, num_shards, index, name)
  
imports: from tensorflow.python.data.ops import dataset_ops
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import ops
  from tensorflow.python.ops import gen_dataset_ops
label_desc: Use image node_id 1 for calling a global function with example usage: _shard(input_dataset, num_shards, index, name) and returns: _ShardDataset