Arm backend: Add support for 5D tensors

[oscarandersson8218]

Updates memory-format handling to consider 5D tensors. Adds a guard to not partition nodes with ranks > 5. This is because the current memory format pass does not handle such ranks in a good way.

cc @digantdesai @freddan80 @per @zingo

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❌ 1 New Failure, 3 Unrelated Failures

As of commit c953f24 with merge base aaf0ecc ():

NEW FAILURE - The following job has failed:

• trunk / unittest-nxp-neutron / linux-job (gh)
  RuntimeError: Command docker exec -t dcfb3b0c73741c9dace4f5176160fe126c4df914eafda60b0b7da6621feb4397 /exec failed with exit code 1

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 / linux / linux-job (gh) (trunk failure)
  examples/models/llama/tests/test_export_llama_lib.py::ExportLlamaLibTest::test_has_expected_ops_and_op_counts
• pull / unittest-editable / linux / linux-job (gh) (trunk failure)
  examples/models/llama/tests/test_export_llama_lib.py::ExportLlamaLibTest::test_has_expected_ops_and_op_counts
• trunk / unittest-release / linux / linux-job (gh) (trunk failure)
  examples/models/llama/tests/test_export_llama_lib.py::ExportLlamaLibTest::test_has_expected_ops_and_op_counts

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

[zingo]

unittest fails are unrelated (and on main also)