nx_f32_conv_transpose2d.nx
buildroot/runtime/nx_f32_conv_transpose2d.nx
about
nx_f32_conv_transpose2d.nx -- software-f32 2D TRANSPOSED convolution / deconv (NCHW), a STANDARD vision op the
f32 tower lacked (only a 1-D conv_transpose in nx_vocops). It is the learned-upsampling head of a SimpleBaseline/
ViTPose pose net (3 stride-2 deconvs 8x-upsample the low-res features into full heatmaps), and is broadly used by
segmentation/generative decoders. Mechanically it is the SCATTER dual of a conv: each input pixel scatter-adds
its value * kernel into a strided output window. Weight layout is PyTorch ConvTranspose2d: [C_in, C_out, KH, KW].
OH = (H-1)*stride - 2*pad + KH + out_pad. Composes ONLY nx_f32_mul / nx_f32_add. license_tier: ORIGINAL
dependencies 2 imports · 1 importers
imports: nx_syscalls.nxnx_f32.nx
imported by: nx_f32_conv_transpose2d_gate.nx
structs
| none |
consts
| 10 | const NX_CT_OK: i64 = 0 |
| 11 | const NX_CT_ERR_ARGS: i64 = 4 |
functions
| 13 | func nx_f32_conv_transpose2d(input: *i64, N: i64, C_in: i64, H: i64, W: i64, |