ns for fx: make input logging work for multi node subgraphs (#54327)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/54327
Makes input logging work properly for multi-node subgraphs.
Test Plan:
```
python test/test_quantization.py TestFXNumericSuiteCoreAPIs.test_linear_fp16_shadow_activations
```
Imported from OSS
Reviewed By: hx89
Differential Revision: D27190137
fbshipit-source-id: 3f39bfd5112d5ee92c1e66c133e970c28db40d46