Implement dumping operator distribution for TOSA graph

[Erik-Lundell]

When dumping the operator distribution after ArmTeste.partition, operators are counted both in the Edge IR and the TOSA graph. The complete output looks something like:
"operators: {op1: 3, op2: 5....}
TOSA operators: {op1: 2, op2: 3}"

Change-Id: I946e8487ad185d9994049ddcdcf7b08153c2597b

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/4970

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

✅ No Failures

As of commit dafdfaa with merge base 7b3549b ():
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[digantdesai]

Nit: perhaps put example output in the commit message or PR summary?

[facebook-github-bot]

@digantdesai has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

[facebook-github-bot]

@digantdesai has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

[Erik-Lundell]

Nit: perhaps put example output in the commit message or PR summary?

Updated the PR summary for both comments :)

[digantdesai]

Seems like ci job test_debug_feats.py::test_dump_tosa_ops is still failing?

[Erik-Lundell]

Seems like ci job test_debug_feats.py::test_dump_tosa_ops is still failing?

Seems like I updated the output message without updating the test, sorry about that. Should be corrected now.

[facebook-github-bot]

@digantdesai has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

[dbort]

@Erik-Lundell @digantdesai The pull / unittest-arm (buck2) / linux-job and trunk / test-arm-reference-delegation / linux-job jobs started failing in trunk (dashboard) after this PR merged.

Example failure (job):

FAILED ../../dev/misc/test_debug_feats.py::test_fail_dump_tosa_ops - SystemExit: 2
= 1 failed, 434 passed, 73 skipped, 134 xfailed, 169 warnings in 467.22s (0:07:47) =

[dbort]

@Erik-Lundell I've reverted this PR in #5131