Public
  1. Public

onnxruntime

Public

Commits

Tianlei WuGitHub
b4457ecb7ae
Tianlei Wu authored and GitHub committed 5d3b2170392
Update Attention operator for GPT2 (#3474)

Add unidrectional mask for Attention operator.
Update mask_index to mask broadcast from B->BxS->BxNxSxS to B->BxSxS->BxNxSxS.