Skip to content

sdpa: support attn_mask.requires_grad, support expanded number of heads in attn_mask #2036

sdpa: support attn_mask.requires_grad, support expanded number of heads in attn_mask

sdpa: support attn_mask.requires_grad, support expanded number of heads in attn_mask #2036