Make `xla::TypeToShape` could convert `stablehlo.token` to xla token shape rather than an empty shape.
Make `xla::TypeToShape` could convert `stablehlo.token` to xla token shape rather than an empty shape.

Also fixed some headers.
