Commit 47a0077
authored
Fix attention fusion in conformer encoder (#23711)
### Description
This PR updates the attention fusion for conformer-encoder models. It is
a follow-up to [this
PR](#23528).
### Motivation and Context
Subsequent modeling code updates have changed (and will continue to
change) the graph fusions. However, the three ending attention mask
nodes (`Cast --> Unsqueeze --> Equal`) will remain. Thus, the attention
fusion should work regardless of any future modeling code changes when
handling the attention mask.1 parent c7aa9a7 commit 47a0077
File tree
1 file changed
+3
-3
lines changed- onnxruntime/python/tools/transformers
1 file changed
+3
-3
lines changedLines changed: 3 additions & 3 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
79 | 79 | | |
80 | 80 | | |
81 | 81 | | |
82 | | - | |
83 | | - | |
| 82 | + | |
| 83 | + | |
84 | 84 | | |
85 | 85 | | |
86 | | - | |
| 86 | + | |
87 | 87 | | |
88 | 88 | | |
89 | 89 | | |
| |||
0 commit comments