Support of transposed convolutions through pixel padding

Xilinx / finn

Dataflow compiler for QNN inference on FPGAs

https://xilinx.github.io/finn

BSD 3-Clause "New" or "Revised" License

708 stars 225 forks source link

Support of transposed convolutions through pixel padding #835

Closed hleblevec closed 8 months ago

hleblevec commented 1 year ago

Gradually adding classes and functions to support the "pixel padding" implementation of transposed convolutions.

[x] Custom_op for PixelPadding nodes
[x] Test bench to validate the full deconvolution (PixelPadding +Conv) [Ongoing]
[x] InferPixelPadding transformation to convert ONNX ConvTranspose node to FMPadding_Pixel + (optional) FMPadding_Batch + ConvInpGen + MVAU
[x] Test bench using InferPixelPadding transformation instead of manual graph creation.

hleblevec commented 8 months ago

Thanks @auphelia for taking the time to review this and sorry if there were things I forgot. I don't know which tests you are talking about but don't hesitate to ask if you need help running them.