xla

ffcc94c7 - Do custom call in _propagate_xla_data