search

uTensor / utensor_cgen

C++ code generator for uTensor https://utensor-cgen.readthedocs.io/en/latest/
Apache License 2.0
50 stars 40 forks source link

Feature request Average Pooling and Min Pooling support #134

Closed victorromeo closed 3 years ago

victorromeo commented 3 years ago

MinPooling appears to be only partially implemented. Avg Pooling is not implemented at all. May we add? Thanks

victorromeo commented 3 years ago

See pull request #135

uTensor implementation still in progress

dboyliao commented 3 years ago

Just to correct your statement a bit, AvgPool and MinPool is supported in uTensor runtime at develop branch.

https://github.com/uTensor/uTensor/blob/develop/src/uTensor/ops/Convolution.hpp#L287-L294

And yes, it haven't added to utensor_cgen, which is different from uTensor runtime but a code generator for the runtime.

dboyliao commented 3 years ago

PR #135 merged. Closing issue.