nir: Add a find_variable_with_[driver_]location helper