def linf_bwd

bendtorch@1.3.0.0 · main.bend:715

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

715def linf_bwd(+n: Nat, +i: Nat, +o: Nat, +par: Nat, p: Lin<i, o>, x: TA.Mat<n, i>, dy: TA.Mat<n, o>, g: Lin<i, o>) -> Bwd<Lin<i, o>, TA.Mat<n, i>>:
716  match p g:
717    case Lin{w, b} Lin{gw, gb}:
718      linf_b3(n, i, o, w, b, gb, TA.Mat.matmul_tn_acc(o, n, i, par, dy, x, gw))

Uses (6)

Used by (1)

bendtorch@1.3.0.0/main:LinearFirst