code: def load_dataset(filename):
      """Reads a file in the `.tfrecord` format.
  
      Args:
        filename: Name of the file containing `tf.train.Example` objects.
  
      Returns:
        An instance of `tf.data.TFRecordDataset` containing the `tf.train.Example`
        objects.
      """
      return tf.data.TFRecordDataset([filename])
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import os
  import threading
  from absl import app
  from absl import flags
  from absl import logging
  import attr
  import neural_structured_learning
  from research.carls import dynamic_embedding_config_pb2
  from research.carls import dynamic_embedding_neighbor_cache
  from research.carls import graph_regularization
  from research.carls import kbs_server_helper_pybind
  import tensorflow
  from google.protobuf import text_format
label_desc: Use image node_id 2 for calling a global function with example usage: load_dataset(filename) and returns: tf