def lnf_w
bendtorch@1.3.0.0 · main.bend:1722
import bendtorch@1.3.0.0/main.bend as Bendtorch
import Base
1722
1723def lnf_w(k: Nat, +j: U32, +off: U32, +mu: F32, +r: F32, y: Array<F32>, rx: Array<F32> & F32, rg: Array<F32> & F32, rb: Array<F32> & F32) -> R4:
1724 match k rx rg rb:
1725 case 0n Tuple{x, +a} Tuple{g, +gv} Tuple{b, +bv}:
1726 R4{x, g, b, Array.set(F32, y, (off + j : U32), ((((a - mu : F32) * r : F32) * gv : F32) + bv : F32))}
1727 case 1n+q Tuple{x, +a} Tuple{g, +gv} Tuple{b, +bv}:
1728 lnf_w(q, (j + 1 : U32), off, mu, r, Array.set(F32, y, (off + j : U32), ((((a - mu : F32) * r : F32) * gv : F32) + bv : F32)), Array.get(F32, x, ((off + j : U32) + 1 : U32)), Array.get(F32, g, (j + 1 : U32)), Array.get(F32, b, (j + 1 : U32)))