Skip to content

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

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 #5327