nx_f32_layernorm.nx
buildroot/runtime/nx_f32_layernorm.nx
about
nx_f32_layernorm.nx -- clean importable flat-array software-f32 LayerNorm (per-row normalize over `dim`, then
affine): out = (x-mean)/sqrt(var+eps) * gamma + beta. The ecosystem has nx_layernorm_forward (NxTensor) and a
layernorm_fwd embedded in a gate, but no plain importable flat lib -- this is it, the ViT/BERT encoder building
block. Composes nx_f32_add/sub/mul/div/sqrt. license_tier: ORIGINAL
dependencies 4 imports · 5 importers
imports: nx_syscalls.nxnx_f32.nxnx_f32_cvt.nxnx_f32_div.nx
imported by: nx_bert_ce_lib.nxnx_f32_bricks_hw_gate.nxnx_f32_layernorm_lib_gate.nxnx_vit_encoder_layer.nxnx_vitpose_forward.nx
structs
| none |
consts
| none |
functions
| 13 | func nx_f32_layernorm(x: *i64, gamma: *i64, beta: *i64, n_rows: i64, dim: i64, eps: i64, out: *i64) -> i64 |
| 40 | func nx_f32_layernorm_sw(x: *i64, gamma: *i64, beta: *i64, n_rows: i64, dim: i64, eps: i64, out: *i64) -> i64 |