onnxruntime
Attention fusion detect num_heads and hidden_size automatically
#6920
Merged

Attention fusion detect num_heads and hidden_size automatically #6920

tianleiwu
tianleiwu detect num_heads and hidden_size automatically
c4ab3e77
tianleiwu tianleiwu requested a review from yufenglee yufenglee 4 years ago
tianleiwu tianleiwu requested a review from wangyems wangyems 4 years ago
tianleiwu tianleiwu requested a review 4 years ago
tianleiwu tianleiwu marked this pull request as draft 4 years ago
tianleiwu add test
dd3f43b9
tianleiwu tianleiwu marked this pull request as ready for review 4 years ago
wangyems
wangyems approved these changes on 2021-03-10
tianleiwu tianleiwu merged 4884eee6 into master 4 years ago
tianleiwu tianleiwu deleted the tlwu/attention_fusion_pruned_model branch 4 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone