[FX] Fix _replicate_for_data_parallel (#63821)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/63821
Test Plan: Imported from OSS
Reviewed By: suo
Differential Revision: D30502115
Pulled By: jamesr66a
fbshipit-source-id: 0f004f95def6e1ba21ccbeab40cb0a739a0ad20c
