transformers

Fix T5Attention shape mismatch under Tensor Parallelism

#45109

Merged