def ce_loop

bend-ml-tensor-fork@0.1.6.0 · main.bend:360

import bend-ml-tensor-fork@0.1.6.0/main.bend as BendMlTensorFork
import Base

360def ce_loop(k: Nat, +mm1: Nat, +mu: U32, +nf: F32, +i: U32, g: B3<F32, U32, F32>) -> B3<F32, U32, F32>:
361  match k:
362    case 0n:
363      g
364    case 1n+p:
365      ce_loop(p, mm1, mu, nf, (i + 1 : U32), ce_step(mm1, mu, nf, i, g))

Uses (6)

Used by (1)