xla

899a0fa4 - Add missing shape metadata for the last replaced fx node (#6805)