[sparse] Change the outer call to ir.RankedTensorType to make it call .get().
[sparse] Change the outer call to ir.RankedTensorType to make it call .get().

PiperOrigin-RevId: 444426777

google/jax,google/jax,google/jax,google/jax