~/jendhub

def gnb_seg

bendtorch-gpu@1.3.0.0 · main.bend:456

import bendtorch-gpu@1.3.0.0/main.bend as BendtorchGpu
import Base

456# dx[off + j] = (((dy·g - ma) - xh·mb)·r) along one row's segment
457def gnb_seg(k: Nat, +j: U32, +off: U32, +mu: F32, +r: F32, +ma: F32, +mb: F32, dx: Array<F32>, rx: Array<F32> & F32, rd: Array<F32> & F32, rg: Array<F32> & F32) -> GB:
458  match k rx rd rg:
459    case 0n Tuple{x, +a} Tuple{dy, +b} Tuple{g, +c}:
460      +xh = ((a - mu : F32) * r : F32)
461      GB{x, dy, g, Array.set(F32, dx, (off + j : U32), ((((b * c : F32) - ma : F32) - (xh * mb : F32) : F32) * r : F32))}
462    case 1n+q Tuple{x, +a} Tuple{dy, +b} Tuple{g, +c}:
463      +xh = ((a - mu : F32) * r : F32)
464      gnb_seg(q, (j + 1 : U32), off, mu, r, ma, mb, Array.set(F32, dx, (off + j : U32), ((((b * c : F32) - ma : F32) - (xh * mb : F32) : F32) * r : F32)), Array.get(F32, x, ((off + j : U32) + 1 : U32)), Array.get(F32, dy, ((off + j : U32) + 1 : U32)), Array.get(F32, g, (j + 1 : U32)))

Uses (12)

Used by (1)