[TPU] Add tensor parallel support for TPU backend#14075

JingyaHuang wants to merge 42 commits into

huggingface:mainhuggingface/diffusers:mainfrom

JingyaHuang:extend-tp-to-tpuJingyaHuang/diffusers:extend-tp-to-tpuCopy head branch name to clipboard