Disable/set tpu_only for tests required tpu backend#3128

NuojCheng wants to merge 1 commit intomainAI-Hypercomputer/maxtext:mainfrom

chengnuojin-tpu-testAI-Hypercomputer/maxtext:chengnuojin-tpu-testCopy head branch name to clipboard