Put TPUEstimatorSpecs through some of the same validation checks as EstimatorSpecs.
Put TPUEstimatorSpecs through some of the same validation checks as EstimatorSpecs.

PiperOrigin-RevId: 221496146

tensorflow/estimator,tensorflow/estimator