Add back in registration (#94452)
Summary: Need to re-register the underscored function in order to have the op present in predictor. This is because older models have been exported with the underscored version.
Test Plan: See if predictor tests pass?
Reviewed By: cpuhrsch
Differential Revision: D43138338
Pull Request resolved: https://github.com/pytorch/pytorch/pull/94452
Approved by: https://github.com/cpuhrsch
