Попробуй искать по имени тега 'input', потом получать из имени атрибут айди и уже передавать его функции.
Получить атрибут можно таким образом:
web_elements = driver.find_elements_by_tag_name('input')
for element in web_elements:
element_id = element.get_attribute('id')
your_function(element_id)