onnx-simplifier icon indicating copy to clipboard operation
onnx-simplifier copied to clipboard

[BUG] Optimize reshape will add new split OP

Open yaoxingwei opened this issue 1 year ago • 0 comments

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.

yaoxingwei avatar Jul 11 '23 08:07 yaoxingwei