search

tlc-pack / relax

Apache License 2.0
193 stars 58 forks source link

[Op] Group for conv2d, ceil mode for max_pool2d #358

Closed MasterJH5574 closed 1 year ago

MasterJH5574 commented 1 year ago

This PR adds the parameter group for conv2d, as well as the option ceil_mode for max_pool2d. They were omitted in the previous round of PR mainly for simplicity, and are introduced now due to their uses in ML frameworks.

The corresponding tests are added as well.

sunggg commented 1 year ago

Thank you @MasterJH5574 for the PR and @YuchenJin for the review!