xla

0e696fc7 - Change all size node logical_element_type to be compatible with the underlying XLA device. (#4416)

Change all size node logical_element_type to be compatible with the underlying XLA device. (#4416) * change all size node return value to be compatible with the underlying xla device. * fix error * fix linter