[ET-VK][sdpa] Use numerically-stable softmax in attention weights#18460

SS-JIA merged 2 commits into

mainpytorch/executorch:mainfrom

gh/SS-JIA/500/origpytorch/executorch:gh/SS-JIA/500/origCopy head branch name to clipboard