Fix partial_tensor ops.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/77580
Replace process_group with _process_group.
Differential Revision: [D36419464](https://our.internmc.facebook.com/intern/diff/D36419464/)
Approved by: https://github.com/wanchaol, https://github.com/fduwjj