Refactoring of NoPE (#3840)

Summary:

X-link: facebookresearch/FBGEMM#927

Some of the refactoring for iRope w/ quantized KV [D69486380] didn't go into the commited version. This diff reflects the changes of the accepted version.

Differential Revision: D71324793

Author: Aya-ZIbra