TPU Executor cleanups: use non-deprecated API, simplify a couple other blocks, newer flags API
TPU Executor cleanups: use non-deprecated API, simplify a couple other blocks, newer flags API

PiperOrigin-RevId: 486948936

tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo