onnxruntime
Add Attention op for multi-head self attention in BERT
#1984

Merged

Add Attention op for multi-head self attention in BERT #1984

tianleiwu merged 11 commits into master from tlwu/attention

Add Attention op for multi head self attention in BERT

79bbc9e1

tianleiwu requested a review from

ybrnathan 6 years ago

tianleiwu requested a review from

yufenglee 6 years ago

tianleiwu requested a review 6 years ago

Add test cases

9b2b92d4

TMVector commented on 2019-10-03

Move op from kOnnxDomain to kMSDomain.

0b272349

merge master

811d4832

fix test

340901bf

Merge remote-tracking branch 'origin/master' into tlwu/attention

38492887

Merge remote-tracking branch 'origin/master' into tlwu/attention

4308ad50

Add float16 test

d39b9e43

fix cpu build error

b51d970f

yufenglee dismissed these changes on 2019-10-06

handle cuda error

55096ac7

tianleiwu dismissed their stale review via 55096ac7 6 years ago

tianleiwu requested a review from

yufenglee 6 years ago

get last cuda error when failed

d512503c

yufenglee approved these changes on 2019-10-07

tianleiwu closed this 6 years ago

tianleiwu reopened this 6 years ago

tianleiwu merged 7b39f509 into master 6 years ago

tianleiwu deleted the tlwu/attention branch 6 years ago

Reviewers

yufenglee

TMVector

ybrnathan

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

onnxruntime Add Attention op for multi-head self attention in BERT #1984 Merged

Add Attention op for multi-head self attention in BERT #1984

onnxruntime
Add Attention op for multi-head self attention in BERT
#1984

Merged