Fix Dawn nightly: select SDPA attn output by shape, not numel (#20283)#20283

meta-codesync[bot] merged 1 commit into

pytorch:mainpytorch/executorch:mainfrom

JulianCloudNTH:export-D108625761JulianCloudNTH/executorch:export-D108625761Copy head branch name to clipboard