search

daquexian / onnx-simplifier

Simplify your onnx model
Apache License 2.0
3.79k stars 380 forks source link

[BUG] Optimize reshape will add new split OP #301

Open yaoxingwei opened 1 year ago

yaoxingwei commented 1 year ago

Hi, When I use onnxsim to optimize the GPT-NEO model. The reshape subgraph is: image

After optimize, the reshape will not be optimized, and a new split OP will be added: image

It seems the reshape structure is the standard structure will be optimized. If any suggestion for the issue? Thanks.

Model GPT-NEO ONNX model

Thanks.