xla

716f44a9 - Revert "Remove override for native_group_norm since we move the composite ker…" (#2504)