Fix: Incorrect Tensor Size for NestedTensor QKV Transform#2450

yucai-intel wants to merge 2 commits intomainintel/torch-xpu-ops:mainfrom

yucai/mha/nested/fixintel/torch-xpu-ops:yucai/mha/nested/fixCopy head branch name to clipboard