code: def open_sharded_output_tfrecords(exit_stack, base_path, num_shards):
      """Opens all TFRecord shards for writing and adds them to an exit stack.
  
      Args:
        exit_stack: A context2.ExitStack used to automatically closed the TFRecords
          opened in this function.
        base_path: The base path for all shards
        num_shards: The number of shards
  
      Returns:
        The list of opened TFRecords. Position k in the list corresponds to shard k.
      """
      tf_record_output_filenames = [
          "{}-{:05d}-of-{:05d}".format(base_path, idx, num_shards)
          for idx in range(num_shards)
      ]
  
      tfrecords = [
          exit_stack.enter_context(tf.io.TFRecordWriter(file_name))
          for file_name in tf_record_output_filenames
      ]
  
      return tfrecords
  
file_path: DALI/docs/examples/use_cases/tensorflow/efficientdet/dataset/tfrecord_util.py
imports: from six.moves import range
  import tensorflow
label_desc: Use image node_id 8 for calling a global function with example usage: open_sharded_output_tfrecords(exit_stack, base_path, num_shards) and returns: tfrecords