~/jendhub

def fork_bwd

bendtorch@1.3.0.0 · main.bend:2117

import bendtorch@1.3.0.0/main.bend as Bendtorch
import Base

2117def fork_bwd(~P1: Type, ~P2: Type, ~X: Type, ~Y1: Type, ~Y2: Type, ~S1: Type, ~S2: Type, ~add: X -> X -> X, ~la: Layer<P1, X, Y1, S1>, ~lb: Layer<P2, X, Y2, S2>, p: Two<P1, P2>, s: Two<S1, S2>, dy: Two<Y1, Y2>, g: Two<P1, P2>) -> Bwd<Two<P1, P2>, X>:
2118  match p s dy g:
2119    case Two{p1, p2} Two{s1, s2} Two{d1, d2} Two{g1, g2}:
2120      fork_b2(~P1, ~X, ~Y1, ~S1, ~add, ~la, P2, p1, s1, d1, g1, Layer.bwd(P2, X, Y2, S2, lb, p2, s2, d2, g2))

Uses (5)

Used by (1)