xla

Introduce apply_xla_patch_to_nn_linear and test that in a scan

#8739

Merged