def cea_row2

bend-ml-tensor-array@0.1.19.0 · main.bend:455

import bend-ml-tensor-array@0.1.19.0/main.bend as BendMlTensorArray
import Base

455def cea_row2(+mm1: Nat, +o: U32, +label: Nat, +nf: F32, +loss: F32, q: AF) -> L2:
456  match q:
457    case AF{c, +s}:
458      cea_row3(mm1, o, label, s, nf, loss, cea_p(Nat.is_le(label, mm1), o, label, s, c))

Uses (8)

Used by (1)