tfio.experimental.serialization.load_dataset

Uses header file to predict the shape and dtypes of tensors for tf.data.