def lin_f2

bendtorch@1.3.0.0 · main.bend:610

import bendtorch@1.3.0.0/main.bend as Bendtorch
import Base
import bend-ml-tensor-array@0.1.18.0/main.bend as TA

610def lin_f2(+n: Nat, -i: Nat, +o: Nat, b: TA.Mat<1n, o>, r: TA.MMulNT<n, i, o>) -> Fwd<Lin<i, o>, TA.Mat<n, o>, TA.Mat<n, i>>:
611  match r:
612    case TA.MMulNT{x, w, z}:
613      lin_f3(n, i, o, w, x, TA.Mat.add_row(n, o, z, b))

Uses (7)

Used by (3)