tlc-pack / relax

Apache License 2.0
195 stars 58 forks source link

[Op][O2g] Convolution, pooling and image operators #341

Closed MasterJH5574 closed 1 year ago

MasterJH5574 commented 1 year ago

As tracked by #332, this PR is the O2g milestone of the high-level operator introduction plan.

It introduces the NN operators conv2d, max_pool2d and adaptive_avg_pool2d, and image operator resize2d, together with some utility functions that are shared by their StructInfo inference. All ops are well-tested.

MasterJH5574 commented 1 year ago

Depends on #340.