-
Notifications
You must be signed in to change notification settings - Fork 3.5k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Fix padding in pooling op #4738
Conversation
@optima2005 @masahi @anijain2305 @alexgl-github @kevinthesun Can you have a look? |
please add a test case |
How to test correctness of |
It's in |
you can just a test case where pooling has different pad size in each dimension. |
37308c4
to
534b60e
Compare
534b60e
to
a653c3c
Compare
The test case was added. @vinx13, @anijain2305 can you have a look again? |
Thanks @apivovarov @masahi this is merged |
Currently we have a bug in pooling op. The code calculates D, H and W padding but uses only D padding. H and W paddings are ignored.
More on the issue: https://discuss.tvm.ai/t/runtime-failed-for-5d-input/5429
This CR fixes it.