TF: rework XLA generate tests#16866

gante merged 5 commits intohuggingface:mainhuggingface/transformers:mainfrom

gante:gpu_xla_testsgante/transformers:gpu_xla_testsCopy head branch name to clipboard