Summary:
Pull Request resolved: https://github.com/pytorch/torchrec/pull/2205

as per title

Reviewed By: gnahzg, iamzainhuda

Differential Revision: D59336088

fbshipit-source-id: 1614039ef2c8d7958c4e98e1b02588c18b932561