xla

acfb2275 - Revert "change all size node return value to be compatible with the underlying xla device. (#4410)" (#4415)