def tie_b2

bendtorch@1.3.0.0 · main.bend:2194

import bendtorch@1.3.0.0/main.bend as Bendtorch

import Base

2194def tie_b2(~P: Type, ~Q: Type, ~X: Type, ~Y: Type, ~Z: Type, ~S1: Type, ~S2: Type, ~la: Layer<P, X, Y, S1>, ~lb: Layer<Q, Y, Z, S2>, q: Q, s1: S1, s2: S2, gq: Q, r: Bwd<P, Z>) -> Bwd<Two<P, Q>, X>:
2195  match r:
2196    case Bwd{p, dz, gp}:
2197      tie_b3(~P, ~X, ~Y, ~S1, ~la, Q, p, s1, gp, Layer.bwd(Q, Y, Z, S2, lb, q, s2, dz, gq))

Uses (5)

Used by (1)

bendtorch@1.3.0.0/main:tie_bwd