xla

605b46e8 - Do custom call in _propagate_xla_data