Add an optional set_mode() driver_ops handler for setting mode before keys