Hi,
I guess you will need to change the implementation in THNN as well. Also keep in mind that this might break some python part of pytorch that expects the convolution to take less arguments.
If you don’t need all the utilities from THCUNN, it might be simpler to create a cpp extension with your modified function.