Not with the current op interface. You would have to write a new op that takes the kernel size as an input tensor. The kernel implementation can be refactored to use most of the current max_pool implementation.
Closing. Feel free to reopen if this is a feature request worth tracking.
+1
I would like to request that feature.
Most helpful comment
+1
I would like to request that feature.