Support enable_gqa and only support 4D Q, K, and V#2558

titaiwangms merged 1 commit intomicrosoft:mainmicrosoft/onnxscript:mainfrom

titaiwangms:titaiwang/support_enable_gqatitaiwangms/onnx-script:titaiwang/support_enable_gqaCopy head branch name to clipboard