nir: add support for counting AoA uniforms in nir_shader_gather_info()