[Dataset] Add optional schema param to read_tfrecords()

[scottjlee]

In cases where a TFRecord contains partially or fully missing feature values, we cannot infer the data type for these features. We should add the option to specify a TensorFlow schema which can explicitly specify the types for such features.