code wiki / (root) / nx_f32_conv2d_backward_fast.nx

nx_f32_conv2d_backward_fast.nx

buildroot/runtime/nx_f32_conv2d_backward_fast.nx

4721 B109 linesdepth 8pulls 13 transitivereach 2 importersview sourcekind librarytopic f32
docsdependenciesstructsconstsfunctions

about

nx_f32_conv2d_backward_fast.nx -- FAST conv2d backward via im2col + the fork-parallel nx_f32_matmul_t, the companion to nx_f32_conv2d_fast so a full TRAINING STEP is fast (the serial backward is the bottleneck ~2/3 of a pose-net step). Formulation (N=1): colT[K,M] = transposed im2col of input; dWeight[C_out,K] = matmul_t(dOut[C_out,M], colT[K,M]); dCol[M,K] = matmul_t(doutT[M,C_out], weightT[K,C_out]); dInput = col2im(dCol); dBias[co]=sum_p dOut. Bit-exact vs nx_f32_conv2d_backward on integer values. ⚠N=1 (dWeight is assigned by matmul_t, not accumulated). license_tier: ORIGINAL

dependencies 3 imports · 2 importers

nx_syscalls.nx nx_f32.nx nx_f32_matmul_t.nx nx_f32_conv2d_backward_fast.nx nx_f32_conv2d_backward_fast_gate.n nx_pose_student_distill.nx

imports: nx_syscalls.nxnx_f32.nxnx_f32_matmul_t.nx

imported by: nx_f32_conv2d_backward_fast_gate.nxnx_pose_student_distill.nx

structs

none

consts

11const NX_CVBF_OK: i64 = 0
12const NX_CVBF_ERR_ARGS: i64 = 4

functions

14func nx_f32_conv2d_backward_fast(input: *i64, N: i64, C_in: i64, H: i64, W: i64,