code: def load_dataset(tfrecord_path, header_path):
      """Uses header file to predict the shape and dtypes of tensors for tf.data."""
      import yaml
  
      header_file = open(header_path)
      header = yaml.load(header_file, Loader=yaml.FullLoader)
  
      feature_desc = build_feature_desc(header)
      parse_func = functools.partial(
          tf.io.parse_single_example, features=feature_desc
      )
      dataset = tf.data.TFRecordDataset(tfrecord_path).map(parse_func)
  
      return dataset
  
file_path: io/tensorflow_io/python/experimental/serial_ops.py
imports: import functools
  import os
  import tempfile
  import numpy
  import tensorflow
label_desc: Use image node_id 10 for calling a global function with example usage: load_dataset(tfrecord_path, header_path) and returns: dataset