View source on GitHub |
Saves a flat dataset as a tfrecord file, and builds a header file for reloading as dataset.
tfio.experimental.serialization.save_dataset(
dataset, tfrecord_path, header_path
)
Must run in eager mode because it depends on dataset iteration and element_spec.