code wiki / (root) / nx_f32_conv_transpose2d.nx

nx_f32_conv_transpose2d.nx

buildroot/runtime/nx_f32_conv_transpose2d.nx

3679 B86 linesdepth 3pulls 3 transitivereach 1 importersview sourcekind librarytopic f32
docsdependenciesstructsconstsfunctions

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

nx_syscalls.nx nx_f32.nx nx_f32_conv_transpose2d.nx nx_f32_conv_transpose2d_gate.nx

imports: nx_syscalls.nxnx_f32.nx

imported by: nx_f32_conv_transpose2d_gate.nx

structs

none

consts

10const NX_CT_OK: i64 = 0
11const NX_CT_ERR_ARGS: i64 = 4

functions

13func nx_f32_conv_transpose2d(input: *i64, N: i64, C_in: i64, H: i64, W: i64,
called by 1: main calls 2: nx_f32_addnx_f32_mul