search

onnx / optimizer

Actively maintained ONNX Optimizer
Apache License 2.0
650 stars 90 forks source link

fuse pad into avgpool and maxpool #65

Closed luliang2luliang closed 2 years ago

daquexian commented 2 years ago

Nice work! Could you please combine fuse_pad_into_avgpool and fuse_pad_into_maxpool into a single pass? Most code in these two passes are duplicated.

daquexian commented 2 years ago

And it will be very appreciated if you can also write some unit tests for the pass you added.

daquexian commented 2 years ago

Thanks for your great work!