Hi,
I’d like to extend max pooling 2d with a new idea. However, for this I need the extend the forward and backward pass of max pooling. Ideally, I would use the cudnn implementation to compute the forward and backward pass of max pooling, but as far as I can see, these are not exposed in the python API. I’ve been looking through the source code, but I’m unable to find the implementation of max_pooling2d. In short: Is there a way to expose the forward and backward pass implementation to the python api? And related: how is max_pool2d implemented, as far as I understand the functions are automatically generated but further than that I’m very confused.
Thanks.