Fixes to basic_input_generator.py to correctly compute the GlobalBatchSize() when there are many tasks per host.
Fixes to basic_input_generator.py to correctly compute the GlobalBatchSize() when there are many tasks per host.

PiperOrigin-RevId: 305193172

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