[pat] Support fixed head_dim in attention groupers - #4982

[pat] Support fixed head_dim in attention groupers#4982

lisjin merged 2 commits into

gh/lisjin/1/basepytorch/ao:gh/lisjin/1/basefrom

gh/lisjin/1/headpytorch/ao:gh/lisjin/1/headCopy head branch name to clipboard