transformers

fc1d841b - change shape to support dynamic batch input in tf.function XLA generate for tf serving (#18372)

change shape to support dynamic batch input in tf.function XLA generate for tf serving (#18372) * change shape to support dynamic batch input in tf.generate * add tests Co-authored-by: nlpcatcode <nlpcodecat@gmail.com>