Arm backend: Support permute-removal for TOSA ops

[oscarandersson8218]

Adds a TOSA specific variant of RemovePermutesAroundElementwiseOps that makes sure that elementwise TOSA backend dialect operators also are covered by this pass. As of now this includes TABLE and RESCALE.

cc @digantdesai @freddan80 @per @zingo @mansnils @Sebastian-Larsson @robell

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❗ 1 Active SEVs

There are 1 currently active SEVs. If your PR is affected, please view them below:

• Ubuntu services are down

❌ 2 Cancelled Jobs, 3 Unrelated Failures

As of commit b765035 with merge base 69989b7 ():

CANCELLED JOBS - The following jobs were cancelled. Please retry:

• Check Labels (gh)
• periodic (gh)

BROKEN TRUNK - The following jobs failed but were present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

• pull / unittest / macos / macos-job (gh) (trunk failure)
  ##[error]The operation was canceled.
• pull / unittest-editable / macos / macos-job (gh) (trunk failure)
  ##[error]The operation was canceled.
• trunk / unittest-release / macos / macos-job (gh) (trunk failure)
  ##[error]The operation was canceled.

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

[mcremon-meta]

I think you can instantiate earlier and add ops, like:

remove_permute_around_elementwise_ops_tosa = RemovePermutesAroundElementwiseOps()
remove_permute_around_elementwise_ops_tosa.permutable_ops | {
    exir_ops.backend.tosa.RESCALE.default,
    exir_ops.backend.tosa.TABLE.default,
}