CodeGen: Use Register

GitOrigin-RevId: 4dad4914f7d5232639e64ffbc816078184f1462c
14 files changed