Arm backend: Add 6D tensor and pixel shuffle/unshuffle support

[pytorchbot]

Adds 6D tensor support required by pixel_shuffle/pixel_unshuffle when given 4D inputs, which means for now we only support 4D inputs. Adds TOSA, VGF and xfailing Ethos-U85 unit tests.

cc @digantdesai @freddan80 @per @zingo @oscarandersson8218

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❌ 4 New Failures, 2 Cancelled Jobs, 4 Unrelated Failures

As of commit 0300e22 with merge base e0dda90 ():

NEW FAILURES - The following jobs have failed:

• Lint / lintrunner / linux-job (gh)
  >>> Lint for backends/arm/_passes/to_tosa_memory_format_pass.py:
• Lint / lintrunner-mypy / linux-job (gh)
  >>> Lint for backends/arm/_passes/to_tosa_memory_format_pass.py:
• pull / unittest-arm-backend-with-no-fvp (test_pytest_models) / linux-job (gh)
  RuntimeError: Command docker exec -t 26f6c34a6f860b036b45fa0e9842a38c302d54d806dd751cdab23d6181308be6 /exec failed with exit code 1
• pull / unittest-arm-backend-with-no-fvp (test_pytest_ops) / linux-job (gh)
  RuntimeError: Command docker exec -t bc3c7e364d4538be95027a9fd83655f04a581889d56459f59043afd938c840be /exec failed with exit code 1

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

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

FLAKY - The following jobs failed but were likely due to flakiness present on trunk:

• pull / test-binary-size-linux / linux-job (gh) (detected as infra flaky with no log or failing log classifier)
• pull / test-build-wasm-linux / linux-job (gh) (detected as infra flaky with no log or failing log classifier)
• pull / test-models-linux (emformer_transcribe, portable, linux.2xlarge) / linux-job (gh) (detected as infra flaky with no log or failing log classifier)
• pull / test-models-linux (mv2, xnnpack-quantization-delegation, linux.2xlarge) / linux-job (gh) (detected as infra flaky with no log or failing log classifier)

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

[GregoryComer]

@digantdesai It looks like there are some issues with the cherry-pick. Can you help fix the PR? It will likely need a maintainer to push changes to the PR branch, so I don't think the ARM engineers can do it. Thanks.

[mansnils]

@digantdesai It looks like there are some issues with the cherry-pick. Can you help fix the PR? It will likely need a maintainer to push changes to the PR branch, so I don't think the ARM engineers can do it. Thanks.

If we for some reason can't get it into 1.0 branch, I think that is ok. It is nice to have but not absolutely needed.

[digantdesai]

Can you help fix the PR?

seems like mypy? let me try..rebasing first.

[digantdesai]

Drop this?

[github-actions]

Looks like this PR hasn't been updated in a while so we're going to go ahead and mark this as Stale.
Feel free to remove the Stale label if you feel this was a mistake.
If you are unable to remove the Stale label please contact a maintainer in order to do so.
If you want the bot to never mark this PR stale again, add the no-stale label.
Stale pull requests will automatically be closed after 30 days of inactivity.

[zingo]

As 1.0 in closed and 1.1 i out lets close this