AITemplate
AITemplate copied to clipboard
[fuse_elementwise] Fix external outputs collection
This PR fixed a corner case where the fuse_elementwise
pass may produce two external output tensors with different shapes. This subgraph pattern is seen the huggingface bert model.