Update TF text classification example
This updates the TF text classification example with several refactors, as well as multi-GPU and TPU support.

It's late so I'd like to do one more pass over everything before merging tomorrow, but I'm opening for reviews before I head off for the evening!