allow customized head_dim#6872

facebook-github-bot merged 5 commits intogh/helunwencser/77/basepytorch/executorch:gh/helunwencser/77/basefrom

gh/helunwencser/77/headpytorch/executorch:gh/helunwencser/77/headCopy head branch name to clipboard