[Torch] Support dynamic head_dim in attention conversion#23680

keshavvinayak01 wants to merge 1 commit intoiree-org:mainiree-org/iree:mainfrom

keshavvinayak01:users/keshavvinayak01/sdpa-dynamic-head-dimkeshavvinayak01/iree:users/keshavvinayak01/sdpa-dynamic-head-dimCopy head branch name to clipboard