nir: Handle vec8/16 in nir_shrink_array_vars