code wiki / (root) / nx_f32_conv2d_grouped.nx

nx_f32_conv2d_grouped.nx

buildroot/runtime/nx_f32_conv2d_grouped.nx

3796 B81 linesdepth 3pulls 3 transitivereach 1 importersview sourcekind librarytopic f32
docsdependenciesstructsconstsfunctions

about

nx_f32_conv2d_grouped.nx -- software-f32 GROUPED 2D convolution (NCHW), generalizing nx_f32_conv2d with a `groups` parameter. groups=1 is the dense conv; groups=C_in (with C_out a multiple) is DEPTHWISE -- the op a MobileNet/EfficientNet-class pose backbone is almost entirely built from, and the single biggest gap the reconnaissance found (nx_f32_conv2d has no groups; the graph-IR has a DEPTHWISE kind with no kernel behind it). Input C_in and output C_out are each split into `groups` contiguous groups; output channel co (in group g=co/(C_out/groups)) convolves ONLY the input channels of group g. Weight layout matches PyTorch grouped conv: [C_out, C_in/groups, KH, KW]. Composes ONLY nx_f32_mul / nx_f32_add. license_tier: ORIGINAL

dependencies 2 imports · 1 importers

nx_syscalls.nx nx_f32.nx nx_f32_conv2d_grouped.nx nx_f32_conv2d_grouped_gate.nx

imports: nx_syscalls.nxnx_f32.nx

imported by: nx_f32_conv2d_grouped_gate.nx

structs

none

consts

11const NX_GCV_OK: i64 = 0
12const NX_GCV_ERR_ARGS: i64 = 4

functions

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