nir: Fix count when we didn't lower load_uniforms but did shift load_ubos.