xla

[scan] Make sure inputs into `fn` are not `device_data` IR nodes

#8769

Merged