Make the behavior similar to existing tfrecord input.
Make the behavior similar to existing tfrecord input.