arange.infer_shape upcast more and be safe again uint64. The old upcast was ok, as the elemw