xla

a6f5b7af - Expand S64 check as the current one does not work (S64 is indeed generally supported on TPU - just not for reduce-window).