Model exporting: Fix the list of input features and drop unsupported collections.
Model exporting: Fix the list of input features and drop unsupported collections.

PiperOrigin-RevId: 212851333

tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor