[SPMD][BE] Generalize factory ops support in SPMD expansion (#99233)
Differential Revision: [](https://our.internmc.facebook.com/intern/diff/)
Differential Revision: [](https://our.internmc.facebook.com/intern/diff/)
Differential Revision: [](https://our.internmc.facebook.com/intern/diff/)
Differential Revision: [](https://our.internmc.facebook.com/intern/diff/)
Differential Revision: [D45028740](https://our.internmc.facebook.com/intern/diff/D45028740)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/99233
Approved by: https://github.com/yifuwang