[nnc] Added a new API to distribute loop and all its parents (#61293)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/61293
Test Plan: Imported from OSS
Reviewed By: ZolotukhinM
Differential Revision: D29560008
Pulled By: navahgar
fbshipit-source-id: e4e459184f20b1872bc242ba8626d0a6df29e810
