NNPACK
NNPACK copied to clipboard
Would you please provide the avg pooling method?
since global avg pooling is essential to the Inception and ResNet architecture,will you implement avg pooling in the future? thank you
In pooling-output.c this file seems to have a bug in line 51.
Const size_t t = x * stride_height + j - padding_left;
should be:
Const size_t t = x * stride_width + j - padding_left;
@liuyi999111 The bug is fixed in 1a75285dab4b76eb3370b46bf985f3bde44d3964. Thank you for reporting.
The mean pooling method looks easy on the basis of the max pooling method. I can help you implement these simple functions.