Explore extending TFRecordInputReader to other data structures

[nthain]

@sorensenjs points out that tf Datasets can read a wide variety of formats. It may be too restrictive to only read in TFRecordFormat.