def gdw_rows

bendtorch-gpu@1.3.0.0 · main.bend:499

import Base

499# sg + dy·xh and sb + dy down one image's rows of a channel, i its cell, by c
500def gdw_rows(k: Nat, +i: U32, +c: U32, +mu: F32, +r: F32, sg: F32, sb: F32, mr: Array<F32>, rx: Array<F32> & F32, rd: Array<F32> & F32) -> GS:
501  match k rx rd:
502    case 0n Tuple{x, +a} Tuple{dy, +b}:
503      GS{x, dy, mr, (sg + (b * ((a - mu : F32) * r : F32) : F32) : F32), (sb + b : F32)}
504    case 1n+q Tuple{x, +a} Tuple{dy, +b}:
505      gdw_rows(q, (i + c : U32), c, mu, r, (sg + (b * ((a - mu : F32) * r : F32) : F32) : F32), (sb + b : F32), mr, Array.get(F32, x, (i + c : U32)), Array.get(F32, dy, (i + c : U32)))