xla

0a1227b5 - Avoid routing special scalar to device data, to give a chance to the XLA compiler to simplify code.