Track node mutations (#152389)
Summary:
X-link: https://github.com/pytorch/pytorch/pull/152389
Approved by: https://github.com/anijain2305
Reviewed By: jeanschmidt
Differential Revision: D74668884
fbshipit-source-id: 355b080f902116c6f12d0fb2b09f3ece26ed92c0