migrate XLA optimization from tf2.15 to 2.20#6

wendi98 wants to merge 3 commits intojoeyye-work:for-serving-2.20joeyye-work/tensorflow:for-serving-2.20from

wendi98:xla-migrationwendi98/tensorflow:xla-migrationCopy head branch name to clipboard