upgrade fuse_concat_into_reshape to support concat->cast->reshape pattern

onnx / optimizer

Actively maintained ONNX Optimizer

Apache License 2.0

650 stars 90 forks source link

upgrade fuse_concat_into_reshape to support concat->cast->reshape pattern #88

Closed HSQ79815 closed 2 years ago

HSQ79815 commented 2 years ago

In the last version , fuse_concat_into_reshape pass only supports the pattern that is Concat -> Reshape.
In this version, upgrade it to support the pattern that is Concat -> Cast -> Reshape.

Signed-off-by: haoshengqiang haoshengqiang79@163.com