Fix type handling for output types from TOSA reference model

[per]

Summary

Fixes handling of other types than float32 when running the TOSA reference model for testing the Arm Backend.

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❌ 2 New Failures

As of commit b67f598 with merge base 03b1ef2 ():

NEW FAILURES - The following jobs have failed:

• trunk / test-demo-backend-delegation (buck2) / linux-job (gh)
  RuntimeError: Command docker exec -t 16922be130bca49600475184bee0836b9b8ff561ea52f8822c8b00754982eed4 /exec failed with exit code 139
• trunk / test-demo-backend-delegation (cmake) / linux-job (gh)
  RuntimeError: Command docker exec -t 18e9cdb91d3191c665a0e55ea500f38319f4bdcf11a7b023e5e045f735b09840 /exec failed with exit code 139

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