Inpatient rehab coding

When patient is admitted to an "acute inpatient" rehab post knee replacement or from acute care hospital, do we use Skilled nursing facility codes (99304-99310) to bill or inpatient hospital e&m codes (99221-99233)?