[PT] Update module partitioner to return parameter node (#101121)
Instead of returning param name, return parameter get_attr node.
Differential Revision: [D45713916](https://our.internmc.facebook.com/intern/diff/D45713916/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/101121
Approved by: https://github.com/angelayi