ns for fx: make shadow input logging work for multi node subgraphs

[vkuzo]

Stack from ghstack:

• ns for fx: rename SugraphTypeRelationship to SubgraphTypeRelationship #55155 ns for fx: rename SugraphTypeRelationship to SubgraphTypeRelationship
• ns for fx: add allowlist for ops with same signature across dtypes #55154 ns for fx: add allowlist for ops with same signature across dtypes
• ns for fx: add linear-relu mod weight extraction #55080 ns for fx: add linear-relu mod weight extraction
• ns for fx: weight extaction for conv1d and conv3d #55079 ns for fx: weight extaction for conv1d and conv3d
• ns for fx: ensure kwargs are handled when graph matching #55078 ns for fx: ensure kwargs are handled when graph matching
• ns for fx: clean up debug print statements #55077 ns for fx: clean up debug print statements
• ns for fx: move it to top level file #55060 ns for fx: move it to top level file
• ns for fx: weight extraction for nni.ConvReLU2d #54335 ns for fx: weight extraction for nni.ConvReLU2d
• ns for fx: make shadow input logging work for multi node subgraphs #54327 ns for fx: make shadow input logging work for multi node subgraphs
• ns for fx: make input logging work for multi-node subgraphs #54326 ns for fx: make input logging work for multi-node subgraphs
• ns for fx: refactor test cases #54280 ns for fx: refactor test cases
• ns for fx: add support for shadowing linear fp16 patterns #54275 ns for fx: add support for shadowing linear fp16 patterns

Summary:

Makes input logging work properly for multi-node subgraphs.

Test Plan:

python test/test_quantization.py TestFXNumericSuiteCoreAPIs.test_linear_fp16_shadow_activations

Reviewers:

Subscribers:

Tasks:

Tags:

Differential Revision: D27190137

[facebook-github-bot]

💊 CI failures summary and remediations

As of commit 2c63d1a (more details on the Dr. CI page):

---

• 1/1 failures possibly* introduced in this PR
  • 1/1 non-scanned failure(s)

---

This comment was automatically generated by Dr. CI (expand for details).

Follow this link to opt-out of these comments for your Pull Requests.

Please report bugs/suggestions to the (internal) Dr. CI Users group.

[facebook-github-bot]

This pull request has been merged in 5319d17.