Arm backend: Fix amax test merge issue#17415

AdrianLundell merged 1 commit intopytorch:mainpytorch/executorch:mainfrom

AdrianLundell:amax_merge_fixAdrianLundell/executorch:amax_merge_fixCopy head branch name to clipboard