def lin_b2

bendtorch@1.3.0.0 · main.bend:630

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

630def lin_b2(+n: Nat, +i: Nat, +o: Nat, par: Nat, b: TA.Mat<1n, o>, x: TA.Mat<n, i>, g: Lin<i, o>, r: TA.MMul<n, o, i>) -> Bwd<Lin<i, o>, TA.Mat<n, i>>:
631  match g r:
632    case Lin{gw, gb} TA.MMul{dy, w, dx}:
633      lin_b3(n, i, o, w, b, dx, gb, TA.Mat.matmul_tn_acc(o, n, i, par, dy, x, gw))

Uses (7)

Used by (1)

bendtorch@1.3.0.0/main:lin_bwd