def lin_b4

bendtorch@1.3.0.0 · main.bend:620

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

620def lin_b4(-n: Nat, -i: Nat, -o: Nat, w: TA.Mat<o, i>, b: TA.Mat<1n, o>, dx: TA.Mat<n, i>, gw: TA.Mat<o, i>, r: TA.MColSum<n, o>) -> Bwd<Lin<i, o>, TA.Mat<n, i>>:
621  match r:
622    case TA.MColSum{dy, gb}:
623      Bwd{Lin{w, b}, dx, Lin{gw, gb}}

Uses (5)

Used by (2)