nir/types: Add a wrapper for count_attribute_slots