def lin_bwd

bendtorch-gpu@1.3.0.0 · main.bend:52

import Base
import bend-ml-tensor-array@0.1.18.0/main.bend as TA
import bend-ml-tensor-gpu@0.1.5.0/main.bend as TG
import bendtorch@1.3.0.0/main.bend as T

52def lin_bwd(+n: Nat, +i: Nat, +o: Nat, p: T.Lin<i, o>, x: TA.Mat<n, i>, dy: TA.Mat<n, o>, g: T.Lin<i, o>) -> T.Bwd<T.Lin<i, o>, TA.Mat<n, i>>:
53  match p:
54    case T.Lin{w, b}:
55      lin_b2(n, i, o, b, x, g, TG.matmul_precise(n, o, i, 0n, dy, w))