Make MultiHeadAttention op return attention probabilities #23125

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed

amancini-N wants to merge 2 commits into microsoft:main from amancini-N:attn-probs-mha

Closed

Make MultiHeadAttention op return attention probabilities #23125

Add CUDA implementation for attn_probs

This check has been archived and is scheduled for deletion. Learn more about checks retention

GitHub Advanced Security / PREfast succeeded Dec 16, 2024 in 10s

No new alerts in code changed by this pull request

View all branch alerts.