flashinfer.norm

归一化层的内核。

rmsnorm(input, weight[, eps, out, enable_pdl])	均方根归一化。
fused_add_rmsnorm(input, residual, weight[, ...])	融合加法均方根归一化。
gemma_rmsnorm(input, weight[, eps, out, ...])	Gemma风格的均方根归一化。
gemma_fused_add_rmsnorm(input, residual, weight)	Gemma风格的融合加法均方根归一化。
layernorm(input, gemma, beta[, eps])	层归一化。