xla

48e12e3d - Fix `non_zero` cast issue (#4243)

Fix `non_zero` cast issue (#4243) * Make nonzero result to reflect the real type * Make sizeNode's type hardware dependent * fix cpu and gpu error * fix gpu and tpu cpp test